[백준] 15663번: N과 M (9) -Python

링크: https://www.acmicpc.net/problem/15663

15663번: N과 M (9)

아이디어

• 백트래킹을 통해 수열 구함
• 방문처리로 본인은 다시 선택하지 못하게 함
• remember 변수로 중복되는 수열이 나오지 않도록 구현

구현

import sys
input = sys.stdin.readline

def dfs():
    if len(temp) == m:
        print(*temp)
        return
    remember = 0
    for i in range(n):
        if not visited[i] and remember != nums[i]:
            visited[i] = True
            temp.append(nums[i])
            remember = nums[i]
            dfs()
            visited[i] = False
            temp.pop()

n, m = map(int, input().split())
nums = sorted(list(map(int, input().split())))
visited = [False] * n
temp = []

dfs()

마무리

• 심심할때마다 돌아오는 N과 M 시리즈
• 머리풀기로 참 좋은듯