Leta Learns

[Python] 백준 15655번 - N과 M (6) 본문

Coding/백준

[Python] 백준 15655번 - N과 M (6)

leta 2022. 2. 25. 21:20

문제 https://www.acmicpc.net/problem/15655

 

15655번: N과 M (6)

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

www.acmicpc.net

 

n과 m (5) 부터는 앞이랑 같은 유형이라 금방 풀 수 있겠지 싶었는데 (6)에서 바로 제동 걸렸다.

 

n개의 자연수를 입력 받고, 오름차순 정렬한 후에 오름차순인 수열을 구하면 되는 간단한 문제였다.

 

한 수열에 같은 수가 나오지 않도록 for문 안의 if문에 조건(i > s)만 넣어주면 되는 것이었는데

갑자기 헷갈려서 시간을 좀 소모했다.

 

import sys
input = sys.stdin.readline

def dfs(s): #s: start
    if len(ans) == m:
        print(*ans)
        return
    
    for i in num:
        if i > s and i not in ans:
            ans.append(i)
            dfs(i)
            ans.pop()

n, m = map(int, input().split())
num = list(map(int, input().split()))
num.sort()
ans = []
dfs(0)

Comments