[BaekJoon] 백준 15654번 : N과 M (5)

15654번 : N과 M (5)

[BaekJoon] 백준 15652번 : N과 M (4) 와 같은 백트래킹을 사용하는 문제이다.

가지를 뻗쳐가며 노드를 계속 방문하다가 조건이 맞지 않으면 부모 노드로 되돌아가 다른 곳을 방문한다.

이 문제는 수열에 순서가 있기때문에 순열로 만들면 된다.

---

import sys


def DFS():
	global n, m, data, stack

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

	for i in range(n):
		stack.append(data[i])

		if len(set(stack)) != len(stack):
			stack.pop()
		else:
			DFS()
			stack.pop()


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

---

import sys
from itertools import permutations


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

for val in res:
	print(*val)

순열을 만들면 되기 때문에 permutation 을 사용해도 된다.