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

 

15654번: N과 M (5)

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

www.acmicpc.net

풀이

백트래킹 N과 M 시리즈 문제입니다. 조합을 구현해주시면 됩니다.

# 15654번 N과 M (5)
import sys
sys.setrecursionlimit(10000)

N, M = map(int, input().split())
nums = list(map(int, input().split()))
nums.sort()
ans = []
visited = [False] * N

def printAns():
    print(*ans)
    return

def choose_num(cur_cnt):
    # 재귀함수 종료조건
    if cur_cnt == M+1:
        printAns()
        return
    
    # 재귀함수 호출
    for  i in range(N):
        if visited[i]:
            continue
        visited[i] = True
        ans.append(nums[i])
        choose_num(cur_cnt+1)
        visited[i] = False
        ans.pop()
        
        
choose_num(1)

출력결과

개발자 성현