[백준/백트래킹] 15652번: N과 M(4) (실버3, python)

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

15652번: N과 M (4)

아이디어

반복문의 범위를 마지막으로 추가된 값부터 N까지로 하고, 마지막으로 추가된 값의 visited를 0으로 변경하면 된다.

전체 코드

N, M = map(int, input().split())

visited = [0] * (N + 1)

def bt(nums, visited):
    if len(nums) == M:
        for num in nums: 
            print(num, end=" ")
        print()
        return
    
    start = nums[-1] if nums else 1
    visited[start] = 0
    for idx in range(start, N + 1):
        if visited[idx] == 0:
            visited[idx] = 1
            bt(nums+[idx], visited)
            visited[idx] = 0
            
bt([], visited)