[백준] 156751 n과 m 3

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

15651번: N과 M (3)

 

python에는 중복순열이 따로 있지 않지만 itertools product로 구현할 수 있다. 

import  itertools 

n,m = map(int, input().split())
perm = [iter for iter in itertools.product(range(1,n+1), repeat=m)]
for p in perm:
    print(' '.join(map(str, p)))

dfs알고리즘을 이용한 코드 

n, m = map(int, input().split())
back = []

def dfs():
    if len(back) == m:
        print(' '.join(map(str, back)))
        return
        
    for i in range(1, n+1):
        back.append(i)
        dfs()
        back.pop()

dfs()

dfs코드가 itertools보다 메모리, 시간 모두 효율적이였는데 특히 메모리에서 큰 차이를 보였다.