BOJ(백준) 15651번 N과 M (3) 파이썬

15651번: N과 M (3)

---

코드

import sys
r=sys.stdin.readline
N,M=map(int,r().split())
l=[i+1 for i in range(N)]
res=[]

def dfs(cnt):
    if cnt==M:
        print(*res)
        return

    for i in range(N):
        res.append(l[i])
        dfs(cnt+1)
        res.pop()

dfs(0)

 

해설

이 문제의 해결 방식 또한 N과 M (1) 문제 풀이와 매우 유사하다. 다만 이 문제에서는 중복을 허용하여 수를 고를 수 있기 때문에 check 와 같은 방문 여부를 확인할 필요가 없다. 단순히 dfs(cnt) 를 재귀적으로 호출해주며 수열의 길이가 M 이 되는 경우에 출력하도록 하였다.

 

결과