백준 문제/백트래킹

15656: N과 M (7)

goldpig 2024. 3. 11. 17:14
728x90

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

 

15656번: N과 M (7)

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

www.acmicpc.net

설명

핵심 아이디어

  • 백트래킹
  • 같은 수를 여러 번 골라도 되기 때문에 for 문에서 조건문을 제거해준다.

코드 구현

import sys
input = sys.stdin.readline

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

def dfs():
	if len(arr) == m:
		print(*arr)
		return
	else:
		for i in data: # 수열이 오름차순
			arr.append(i)
			dfs()
			arr.pop()
dfs()
728x90