코딩테스트 대비/단계별 코딩 테스트 준비(27일 과정)

[백트래킹/Python] 15649번: N과 M(1) - 효과는 굉장했다!

bluetag_boy 2022. 2. 11. 21:28
반응형
 

15649번: N과 M (1)

한 줄에 하나씩 문제의 조건을 만족하는 수열을 출력한다. 중복되는 수열을 여러 번 출력하면 안되며, 각 수열은 공백으로 구분해서 출력해야 한다. 수열은 사전 순으로 증가하는 순서로 출력해

www.acmicpc.net

 

알고리즘 분류

  • 백트래킹

 

 

SOLUTION

 

# 백트래킹을 이용한 풀이

import sys

def dfs():
    # 주어진 개수만큼 채워지면 출력한다
    if len(arr) == M:
        print(*arr)
        return
    
    for num in range(1,N+1):

        if num not in arr:
            arr.append(num)
            dfs()
            arr.pop()


N, M = map(int, sys.stdin.readline().split())
arr = []   
dfs()

 

# permutations 함수를 이용한 풀이

import sys
from itertools import permutations

N, M = map(int, sys.stdin.readline().split())
permu_list = list(permutations([i for i in range(1,N+1)], M))

for i in permu_list:
    print(*i)