[백준] 15664번: N과 M (10) - python

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

15664번: N과 M (10)

 

아이디어

 

재귀함수의 인자값으로 index를 넘겨주어서 중복되는 수열을 제거하는 방식으로 작성

 

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


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

 

완성