[백준 2798] 블랙잭(Feat.Python)[Combination]

문제

카지노에서 제일 인기 있는 게임 블랙잭의 규칙은 상당히 쉽다. 카드의 합이 21을 넘지 않는 한도 내에서, 카드의 합을 최대한 크게 만드는 게임이다. 블랙잭은 카지노마다 다양한 규정이 있다.

한국 최고의 블랙잭 고수 김정인은 새로운 블랙잭 규칙을 만들어 상근, 창영이와 게임하려고 한다.

김정인 버전의 블랙잭에서 각 카드에는 양의 정수가 쓰여 있다. 그 다음, 딜러는 N장의 카드를 모두 숫자가 보이도록 바닥에 놓는다. 그런 후에 딜러는 숫자 M을 크게 외친다.

이제 플레이어는 제한된 시간 안에 N장의 카드 중에서 3장의 카드를 골라야 한다. 블랙잭 변형 게임이기 때문에, 플레이어가 고른 카드의 합은 M을 넘지 않으면서 M과 최대한 가깝게 만들어야 한다.

N장의 카드에 써져 있는 숫자가 주어졌을 때, M을 넘지 않으면서 M에 최대한 가까운 카드 3장의 합을 구해 출력하시오.

 

풀이

• 본인은 이 문제를 보고 처음에는 while문 혹은 3중 for문을 생각했었다.
• 파이썬으로 문제를 풀다보니 시간초과의 두려움은 없었으나 다른 방법을 찾아보려고 하였음
• 파이썬의 combination을 사용하여 문제를 풀고자 하였음

from itertools import combinations

N, target = map(int, input().split())
arr = sorted(list(map(int, input().split())))
answer = 0
for unit in combinations(arr, 3):
    if answer < sum(unit):
        if sum(unit) <= target:
            answer = sum(unit)

print(answer)

 

• 간략하게 설명하면 combination(arr, 3) 을 사용하면 arr안에 있는 원소 3개씩의 모든 조합을 리턴해준다.
• 이를 활용하여 for문에 unit에 하나씩 담아서 각 조합의 합을 비교하는 로직을 구현
• answer에는 현재 target에 가장 가까운 값을 가지고 있음