[백준] 15654번: N과 M (5) - python

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

15654번: N과 M (5)

 

아이디어

 

부분집합을 결국 만드는 문제이므로 check 리스트를 만들어서 방문하지 않았으면 들어가서 상태트티를 생성하고 방문했으면 넘어가고, Level이 m과 같아지면 저장한 값들을 출력.

 

def DFS(L, s):
    if L == m:
        for i in range(m):
            print(res[i], end=" ")
        print()
    else:
        for i in range(n):
            if ch[i] == 0:
                ch[i] = 1
                res[L] = arr[i]
                DFS(L+1, i)
                ch[i] = 0


n, m = map(int, input().split())
arr = list(map(int, input().split()))
res = [0] * (m+1)
ch = [0] * (n+1)
arr.sort()
DFS(0, 0)

 

완성