728x90
반응형
문제 출처:https://www.acmicpc.net/problem/15650
15650번: N과 M (2)
한 줄에 하나씩 문제의 조건을 만족하는 수열을 출력한다. 중복되는 수열을 여러 번 출력하면 안되며, 각 수열은 공백으로 구분해서 출력해야 한다. 수열은 사전 순으로 증가하는 순서로 출력해
www.acmicpc.net
풀이:
n,m = map(int,input().split())
visited = [False] * n
result = []
def solve(num):
if num == m:
print(*result)
return
for i in range(n):
if visited[i] == False:
visited[i] = True
result.append(i + 1)
solve(num+1)
visited[i] = True
result.pop()
for j in range(i+1, n):
visited[j] = False
solve(0)
처음 i의 DFS 탐색에 사용 된 i값을 제외시키는 것이 순열과 다른 조합의 차이다.
풀이2:
from itertools import combinations
N, M = map(int, input().split())
p = combinations(range(1, N+1), M)
for i in p:
print(*i)
반응형
댓글