Algorithm(CodeTree, Python)/Backtracking

[코드트리] k개 중에 1개를 n번 뽑기 Python

kurooru 2023. 1. 27. 11:12
# k, n 입력
k, n = map(int, input().split())

# 함수들
# print_ans()
def print_ans():
    for num in ans_list:
        print(num, end=' ')
    print()

# choose(curr_idx)
def choose(curr_idx):
    
    # 종료조건
    if curr_idx == n + 1:
        # 출력
        print_ans()
        return
    
    # 넣어주기
    for i in range(1, k+1):
        # 넣고
        ans_list.append(i)
        # 다음자리로
        choose(curr_idx + 1)
        # 빼주기
        ans_list.pop()

# 설계
# ans_list
ans_list = []
# choose
choose(1)