백준,N과 M (6),15655

[유형]

백트래킹

 

[문제링크]

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

 

[요약]

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

• N개의 자연수 중에서 M개를 고른 수열
• 고른 수열은 오름차순이어야 한다.

 

[문제풀이]

앞의 N과 M(2),15650번과 다른 조건은 수열이 주어진다는 점이다.

사전 순으로 증가는 순서로 출력하기 위해 sort()함수를 사용한다.

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

def dfs(start,lst):
    # 종료조건
    if len(lst) == M:
        ans.append(lst)
        return

    for j in range(start,N):
        if v[j] == 0:
            v[j] = 1
            dfs(j+1,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(0,[])

for lst in ans:
    print(*lst)