📒 문제
카지노에서 제일 인기 있는 게임 블랙잭의 규칙은 상당히 쉽다. 카드의 합이 21을 넘지 않는 한도 내에서, 카드의 합을 최대한 크게 만드는 게임이다. 블랙잭은 카지노마다 다양한 규정이 있다.
한국 최고의 블랙잭 고수 김정인은 새로운 블랙잭 규칙을 만들어 상근, 창영이와 게임하려고 한다.
김정인 버전의 블랙잭에서 각 카드에는 양의 정수가 쓰여 있다. 그 다음, 딜러는 N장의 카드를 모두 숫자가 보이도록 바닥에 놓는다. 그런 후에 딜러는 숫자 M을 크게 외친다.
이제 플레이어는 제한된 시간 안에 N장의 카드 중에서 3장의 카드를 골라야 한다. 블랙잭 변형 게임이기 때문에, 플레이어가 고른 카드의 합은 M을 넘지 않으면서 M과 최대한 가깝게 만들어야 한다.
N장의 카드에 써져 있는 숫자가 주어졌을 때, M을 넘지 않으면서 M에 최대한 가까운 카드 3장의 합을 구해 출력하시오.
입력
첫째 줄에 카드의 개수 N(3 ≤ N ≤ 100)과 M(10 ≤ M ≤ 300,000)이 주어진다. 둘째 줄에는 카드에 쓰여 있는 수가 주어지며, 이 값은 100,000을 넘지 않는 양의 정수이다.
합이 M을 넘지 않는 카드 3장을 찾을 수 있는 경우만 입력으로 주어진다.
출력
첫째 줄에 M을 넘지 않으면서 M에 최대한 가까운 카드 3장의 합을 출력한다.
✏️ 풀이
N값의 제한이 100으로 매우 낮기 때문에 전수 조사로 카드를 3장 뽑으면 시간복잡도 100만 정도의 연산에서 구할 수 있다. 하지만 for문의 반복을 사용하는 대신, itertools의 combinations를 통해 조합을 구하여 간단히 풀어보았다.
from itertools import combinations
n, m = map(int, input().split())
cards = list(map(int, input().split()))
comb = list(combinations(cards, 3))
result = 0
for i in comb:
if sum(i) == m:
result = m
break
elif sum(i) < m and sum(i) > result:
result = sum(i)
print(result)하지만, combinations를 사용하면 메모리가 40848kb 시간이 172ms로 for문의 3중 반복을 이용한 풀이에 비해 효율성이 떨어지게 된다.
이를 고려한 for문을 이용한 풀이는 다음과 같다.
n, m = map(int, input().split())
cards = list(map(int, input().split()))
result=0
for i in range(0,n):
for j in range(i+1,n):
for k in range(j+1,n):
sum=cards[i]+cards[j]+cards[k]
if sum<=m and sum>result:
result=sum
print(result)이 풀이를 이용하면 메모리 29200kb, 시간 108ms의 결과를 얻을 수 있다.
