중복순열 구하기

문제

• 1~N까지 번호 적힌 구슬
• 중복을 허락하여 M번 뽑아 일렬로 나열하는 방법을 모두 출력

my_solution

from itertools import product

n, m = map(int, input().split())
lists = [i for i in range(1, n + 1)]
cnt = 0

for i in product(lists, repeat=m):
    print(*i, sep=" ")
    cnt += 1
print(cnt)

• 중복을 허락하여 모든 조합을 나열해야하기에 product 사용

soltuion

n, m = map(int, input().split())
res = [0] * m
cnt = 0


def recursive(level):
    global cnt
    if level == m:
        for j in range(m):
            print(res[j], end=" ")
        print()
        cnt += 1
    else:
        for i in range(1, n + 1):
            res[level] = i
            recursive(level + 1)


recursive(0)
print(cnt)

• DFS를 연습하기 위해 재귀적으로 구현한 방법