백준,N과 M (2),15650

[유형]

백트래킹

 

[문제링크]

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

 

[요약]

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

• 1부터 N까지 자연수 중에서 중복 없이 M개를 고른 수열
• 고른 수열은 오름차순이어야 한다.

 

[문제풀이]

앞의 N과 M(1),15649번과 다른 조건은 고른 수열은 오름 차순이어야한다는 것이다.

따라서 start를 추가해서, 첫번째 수가 두번째 수보다 크거나 같지 않도록 코드를 변경한다.

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+1):
        if v[j]==0:
            v[j] = 1
            dfs(j+1,lst+[j])
            v[j] = 0

N, M = map(int, input().split())
ans = []
v = [0] * (N+1)

dfs(1,[])

for lst in ans:
    print(*lst)