백준,N과 M (9),15663

[유형]

백트래킹

 

[문제링크]

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

 

[요약]

N개의 자연수와 자연수 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램을 작성하시오.

• N개의 자연수 중에서 M개를 고른 수열
• 한 줄에 하나씩 문제의 조건을 만족하는 수열을 출력한다. 중복되는 수열을 여러 번 출력하면 안되며, 각 수열은 공백으로 구분해서 출력해야 한다.

 

[문제풀이]

앞의 N과 M(6),15655번과 다른 조건은 중복되는 수열을 여러번 출력하면 안된다는 것이다. 따라서 set을 사용한다. set은 순서가 없기 때문에 sort를 해줘야한다.

import sys
def input():
    return sys.stdin.readline().rstrip()

def dfs(lst):
    # 종료조건
    if len(lst) == M:
        ans.add(tuple(lst))

        return

    for j in range(N):
        if v[j]==0:
            v[j]=1
            dfs(lst+[num_list[j]])
            v[j]=0

N, M = map(int, input().split())
num_list = list(map(int, input().split()))

num_list.sort()

v = [0] * (N+1)
ans = set()
dfs([])


for lst in sorted(ans):
    print(*lst)