[백준] 15664번 N과 M (10)

💡문제접근

• 조합과 정렬을 이용해서 간단히 해결했다.

💡코드(메모리 : 30616KB, 시간 : 56ms)

from itertools import combinations

import sys
input = sys.stdin.readline

N, M = map(int, input().strip().split())
A = list(map(int, input().strip().split()))
# 배열 A 오름차순 정렬
A.sort()
# 조합 순서쌍 중복 제거
li = set(combinations(A, M))
# 사전순으로 오름차순 정렬
li = sorted(li)

for i in li:
    print(*i)

📌백트래킹 풀이 방법(메모리 : 31256KB, 시간 : 44ms)

import sys
input = sys.stdin.readline

N, M = map(int, input().strip().split())
li = list(map(int, input().strip().split()))
li.sort()
visited = [False] * N
res = []

def recursive(start):
    if len(res) == M:
        print(" ".join(map(str, res)))
        return 0
    else:
        flag = 0
        for i in range(start, len(li)):
            if not visited[i] and flag != li[i]:
                visited[i] = True
                res.append(li[i])
                flag = li[i]
                recursive(i+1)
                visited[i] = False
                res.pop()
recursive(0)

💡소요시간 : 2m