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

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

15654번: N과 M (5)

 

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

• N개의 자연수 중에서 M개를 고른 수열

 

앞의 N과 M 문제와 달라진 점은 입력으로 N개의 수가 주어진다는 것이다.

 

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

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

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

 

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

arr: 입력받은 수열

result: 출력할 수열

 

import sys
input = sys.stdin.readline

def go(index, n, m):
    if index == m:
        print(' '.join(map(str, result)))

    for i in range(n):
        if visited[i]:
            continue

        visited[i] = True
        
        # 입력 받은 수를 result에 저장한다
        result.append(arr[i])

        go(index+1, n, m)

        visited[i]=False
        result.pop()

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

# 입력 받은 수를 오름차순으로 정렬한다
arr.sort()

visited = [False] * 10
result = []

go(0, n, m)

 

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

15654번: N과 M (5)