[백준 15654] N과 M (5) - python (solved.ac - 실버 3)

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

15654번: N과 M (5)

풀이

문제를 보았을때 알수있는 힌트

이 문제는 정해진 숫자를 차례대로  출력 될수있게 하면 되는 문제이다. 

백트래킹과 dfs를 이용해서 풀면 쉽게 풀린다. 또한 그냥 itertools의 permutations을 통해 풀면 간단하게 풀수있다.

# dfs와 백트래킹
import sys

input= sys.stdin.readline
def dfs(x,y):
    if m==y:
        print(*x)
    else:
        for i in range(0,n):
            if n_list[i] not in x:
                x.append(n_list[i])
                dfs(x,y+1)
                x.pop()

n,m = map(int,input().split())
n_list = list(map(int,input().split()))
n_list.sort()

for i in range(0,n):
    temp=[n_list[i]]
    dfs(temp,1)

# itertools의 permutations
import sys
from itertools import permutations

input= sys.stdin.readline

n,m = map(int,input().split())

n_list = list(map(int,input().split()))
n_list.sort()
result =list(permutations(n_list,m))
for i in result:
    print(*i)

두가지 방법 모두 결과는 정답으로 나온다.

 

https://github.com/dydwkd486/coding_test/blob/main/baekjoon/baekjoon15654.py

GitHub - dydwkd486/coding_test: 코딩테스트 공부한 내용 정리