Koala - 11기/코딩테스트 준비 스터디

[백준/Python] N과 M (백트레킹)

쥬발이 2023. 7. 16. 23:59

1. 문제 

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

 

15656번: N과 M (7)

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

www.acmicpc.net

 

2. 코드

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

nums.sort()

ans = []

def back():
    if len(ans) == m:
        print(" ".join(map(str,ans)))
        return

    for i in range(n):
        ans.append(nums[i])
        back()
        ans.pop()
back()

 

3. 풀이

재귀를 이용한 백트레킹 알고리즘을 사용하여 풀었다.