본문 바로가기
알고리즘

[백준] 15656번: N과 M (7) - python

by 육빔 2024. 4. 8.
728x90
반응형

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

 

15656번: N과 M (7)

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

www.acmicpc.net

 

계속 시리즈별로 푸니까 이제 상태트리를 좀 더 수월하게 작성이 바로바로 된다.

 

아이디어

 

이번엔 중복이 허용되기에 저장할 인자도 없고 무지성으로 상태트리를 만들어준다.

def DFS(L):
    if L == m:
        for i in range(m):
            print(res[i], end= ' ')
        print()
    else:
        for i in range(n):
            res[L] = arr[i]
            DFS(L+1)

n, m = map(int, input().split())
arr = list(map(int, input().split()))
res = [0] * m
arr.sort()
DFS(0)

 

완성

728x90
반응형