본문 바로가기
백준 알고리즘(BOJ)

백준 알고리즘 15654 (N과 M(5)) - python

by Think_why 2019. 10. 15.

[문제] 백준 알고리즘 15654 (N과 M(5)) - python

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

 

15654번: N과 M (5)

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

www.acmicpc.net

 

이제는 N과 M을 자연수가 아닌 지정된 배열을 출력시킨다.

중복 수열을 허용하지 않고, 사전순은 동일하다.

 

그렇다면, 출력용 배열인 out에 i가 아닌 배열의 [i]번째를 넣으면 동일한 구조이다.

사전순으로 출력하기 위해서 지정된 배열 L을 sort() 시키고,

out에 L[i]를 append 해주어 어렵지 않게 풀 수 있었다.

 

[Code]

N, M = map(int, input().split())
L = list(map(int, input().split()))

L.sort()
visited = [False] * N
out = []

def solve(depth, N, M):
    if depth == M:
        print(' '.join(map(str, out)))
        return
    for i in range(N):
        if not visited[i]:
            visited[i] = True
            out.append(L[i])
            solve(depth+1, N, M)
            out.pop()
            visited[i] = False

solve(0, N, M)
728x90

댓글