백준,N과 M (4),15652

[유형]

백트래킹

 

[문제링크]

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

 

[요약]

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

• 1부터 N까지 자연수 중에서 M개를 고른 수열
• 같은 수를 여러 번 골라도 된다.
• 고른 수열은 비내림차순이어야 한다.

[문제풀이]

앞의 N과 M(2),15650번과 다른 조건은 고른 수열은 비내림차순이라는 것과 같은 수를 여러 번 골라도 된다는 것이다.

따라서 가지치기 조건을 삭제하고, 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):
            dfs(j,lst+[j])

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

dfs(1,[])

for lst in ans:
    print(*lst)