백준,N과 M (5),15654

[유형]

백트래킹

 

[문제링크]

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

 

[요약]

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

• N개의 자연수 중에서 M개를 고른 수열

 

[문제풀이]

앞의 N과 M(1),15649번과 다른 조건은 수열이 주어진다는 것이다. 

따라서 num_list를 넘겨주고 이미 선택된 자연수를 선택하지 않기 위해 가지치기 조건을 사용한다.

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

def dfs(lst):
    # 종료조건
    if len(lst) == M:
        ans.append(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()

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

for lst in ans:
    print(*lst)