[BaekJoon] 백준 15666번 : N과 M (12)

15666번 : N과 M (12)

[BaekJoon] 백준 15657번 : N과 M (8) 같은 문제인데 배열을 직접 입력한다는 부분이 다른 문제이다.

조건에서 같은 수를 여러번 골라도 되기 때문에 중복 값들은 전부 없앤 뒤 조합을 구하면 된다.

조합은 순서가 상관 없고 조건에서 비내림차순으로 출력하라고 했기 때문에 뒤에 오는 값이 앞에 오는 값보다 작으면 안된다.

이 조건들을 다 지켜서 조합을 만들면 된다.

물론 combination_with_replacement 를 사용해도 된다.

---

import sys


def DFS():
	if len(stack) == m:
		print(*stack)
		return

	for i in range(len(data)):
		if stack and stack[-1] > data[i]:
			continue

		stack.append(data[i])
		DFS()
		stack.pop()


n, m = map(int, sys.stdin.readline().rsplit())
data = sorted(list(set(map(int, sys.stdin.readline().rsplit()))))
stack = []
DFS()

---

import sys
from itertools import combinations_with_replacement


n, m = map(int, sys.stdin.readline().rsplit())
data = sorted(list(set(map(int, sys.stdin.readline().rsplit()))))

for val in combinations_with_replacement(data, m):
	print(*val)