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

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

15649번: N과 M (1)

 

브루트포스 방법을 사용하여 뽑기 문제를 풀 때 유형은 두 가지로 나뉜다.

 

1. 순서: N개 중에서 M개를 뽑을 때, 순서를 고려하는 경우

2. 선택: N개 중에서 M개를 뽑을 때, 순서를 고려하지 않는 경우

 

이 문제는 순서를 고려하지 않는다. 

 

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

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

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

 

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

arr: 수열을 담는 배열

 

import sys
input = sys.stdin.readline

def go(index, n, m):
    # 종료조건: m개를 다 골랐다면 수열 출력
    if index == m:
        print(' '.join(map(str, arr)))
        return

    for i in range(0, n):
        # 사용: T, 미사용: F
        if c[i]: 
            continue # 사용한 것이라면 넘기기

        visited[i] = True # 사용
        arr.append(i+1) # arr 입력
        go(index+1, n, m)
        visited[i] = False
        arr.pop()

    
n, m = map(int, input().split())
visited = [False] * 10
arr =[]

# 1부터 N까지 자연수 중에서 중복 없이 M개를 고른 수열
go(0, n, m);