[백준] 15657번: N과 M (8) - python

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

15657번: N과 M (8)

 

이상하게 시리즈가 올라갈수록 쉬운거같은 느낌이..

 

아이디어

 

상태트리에 n개중에 m개를 뽑는데 인덱스를 저장해가면서 인덱스~n까지 상태트리를 만들고 Level이 m에 닿으면 출력해주는 로직.

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

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

 

완성