[백준/브루트포스] 15655 N과 M (6) (Python, 파이썬)

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

15655번: N과 M (6)

 

 

이 문제의 조건은 다음과 같다.

• N개의 자연수 중에서 M개를 고른 수열
• 고른 수열은 오름차순이어야 한다.

 

재귀함수를 이용하여 풀었는데, 이 때 재귀함수는 어떤 위치에 올 수 있는 수를 결정하는 용도이다.

재귀함수에서 중요한 것은 종료조건이다.

여기서의 종료조건은 M개를 다 뽑았을 때이다. 그때 배열에 저장한 수를 모두 출력하면 된다. 

 

visited: 해당 수를 사용했는지 안했는지 판별하는 배열(사용: True, 미사용: False)

arr: 입력받은 배열

result: 출력할 배열

 

start는 오름차순으로 result에 저장하기 위하여 만들었다.

 

 

import sys
input = sys.stdin.readline

def go(index, start, n, m):
    if index == m:
        print(' '.join(map(str, result)))
        return 
    
    for i in range(start, n):
        if visited[i]:
            continue

        visited[i] = True
        result.append(arr[i])

        go(index+1, i+1, n, m)

        result.pop()
        visited[i] = False

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

result = []
visited=[False] * n

go(0, 0, n, m)