Problem Solving/백준
[BOJ 백준] 15655번 : N과 M(6) - Python
돌돌김
2021. 1. 3. 00:52
15655번: N과 M (6)
N개의 자연수와 자연수 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램을 작성하시오. N개의 자연수는 모두 다른 수이다. N개의 자연수 중에서 M개를 고른 수열
www.acmicpc.net
조합을 구하는 문제이다.
순열과 달리, 사용했던 원소를 체크해주는게 관건이다.
# 조합
n, m = map(int, input().split())
data = list(map(int, input().split()))
data.sort()
result = []
checked = [False]*n
def dfs(idx, count):
if count == m:
print(*result)
return
for i in range(idx, len(data)):
if checked[i] == True: continue
checked[i] = True
result.append(data[i])
dfs(i, count+1)
result.pop()
checked[i] = False
dfs(0, 0)