Competition/Baekjoon

[백준][파이썬] 15654 N과 M (5) 풀이

bisi 2022. 10. 22. 14:31
문제 출처 

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

 

15654번: N과 M (5)

N개의 자연수와 자연수 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램을 작성하시오. N개의 자연수는 모두 다른 수이다. N개의 자연수 중에서 M개를 고른 수열

www.acmicpc.net

 

 

 

접근 방식 및 풀이

- 정렬 로직은 sorted 함수 사용

- DFS 로 재귀로 부르면서 list에 담다가 길이가 M이면 return

 

 

 

 

소스 코드 
N,M = map(int, input().split())
arr = list(map(int, input().split()))

tmp_list = []

def dfs():

    if len(tmp_list) == M:
        for i in range(M):
            print(tmp_list[i], end=' ')
        print('')
        return

    for i in range(0,N):

        if arr[i] in tmp_list:
            continue

        tmp_list.append(arr[i])
        dfs()
        tmp_list.pop()


arr = sorted(arr)
dfs()

 

 

결과