[백준] N과 M (1)

문제

https://www.acmicpc.net/problem/15649

15649번: N과 M (1)

문제
자연수 N과 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램을 작성하시오.

1부터 N까지 자연수 중에서 중복 없이 M개를 고른 수열
입력
첫째 줄에 자연수 N과 M이 주어진다. (1 ≤ M ≤ N ≤ 8)

출력
한 줄에 하나씩 문제의 조건을 만족하는 수열을 출력한다. 중복되는 수열을 여러 번 출력하면 안되며, 각 수열은 공백으로 구분해서 출력해야 한다.

수열은 사전 순으로 증가하는 순서로 출력해야 한다.

 

예제 입력 1 
3 1

예제 출력 1 
1
2
3

---

풀이

import sys
input = sys.stdin.readline
from itertools import permutations

n, m = map(int, input().split())
list_n = [i for i in range(1, n+1)]

answers = list(permutations(list_n, m))

for i in answers:
    print(' '.join(map(str, i)))

 

---

설명

처음에 보고 순열과 조합이 떠오르긴 했으나 정확한 코드식을 까먹었다..ㅠ

 

1. 순열: 서로 다른 원소의 순서를 고려해서 나열.
from itertools import permutations

permutations(배열, 뽑는 개수)

▼ 집합 1,2,3,4 에서 2개를 뽑는 순열. 튜플() 로 출력하므로, 결과를 볼때는 list, tuple 과 같이 써야지 보는게 가능함

from itertools import permutations

a = [1,2,3,4]
res = permutations(a,2)
print(list(res))	# list 와 같이 써야 결과 보는게 가능

pycharm 에서 확인한 permutations 결과

 

2. 조합: 서로다른 원소의 순서를 고려하지 않고 나열

from itertools import commbinations

combinations(배열, 뽑는 개수)

from itertools import combinations

a = [1,2,3,4]
res = combinations(a,2)
print(list(res))	# list 와 같이 써야 결과 보는게 가능

pycharm 에서 확인한 combinations 결과

 

3. 코드에서 헷갈렸던 사항들

문제 1) split 으로 잘라와서 permutations 함수에 넣는것은 불가능함!

import sys
input = sys.stdin.readline
from itertools import permutations

n, m = list(map(int, input().split()))

print(list(permutations(n,m)))

위와 같은 문제가 발생한다..!

문제 2) [(1,), (2,), (3,)]  <-이런식으로 정답이 나왔었다..

import sys
input = sys.stdin.readline
from itertools import permutations

n, m = map(int, input().split())
list_n = [i for i in range(1, n+1)]

print(list(permutations(list_n, m)), end='')

 

 

문제 3) (1,) (2,) (3,) 이런식으로 출력 됨

import sys
input = sys.stdin.readline
from itertools import permutations

n, m = map(int, input().split())
list_n = [i for i in range(1, n+1)]

answers = [permutations(list_n, m)]

for i in answers:
    print(' '.join(map(str, i)))

문제 4) 정답~~

import sys
input = sys.stdin.readline
from itertools import permutations

n, m = map(int, input().split())
list_n = [i for i in range(1, n+1)]

answers = list(permutations(list_n, m))

for i in answers:
    print(' '.join(map(str, i)))