1. 문제

https://www.acmicpc.net/problem/2798

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


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


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


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


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

2. Key point

🍕 조합 Combination (nCr)

조합이란 n개의 원소 중 순서에 관계 없이 r개의 원소를 추출하여 배열로 나타내는 것을 말한다. 순서에 관계 없기 때문에 원소의 종류가 같으면 같은 배열로 나타낸다.

예를 들어, [1,2,3]이라는 리스트에서 2개의 원소를 추출하여 조합을 만들면,
[(1,2), (1,3), (2,3)]으로 나타낼 수 있다.
→ (1,2)와 (2,1)은 같은 것으로 여기기 때문에 중복하지 않는다.

파이썬에서는 itertools에 있는 combinations 함수를 통해 조합을 구할 수 있다.

from itertools import combinations
a = [1,2,3]
combi_a = combinations(a, 2)

print(combi_a) 
# <itertools.combinations object at 0x000001466ED25CC0>

print(list(combi_a))	
# [(1, 2), (1, 3), (2, 3)]

3. 나의 풀이

(1) 첫번째 풀이 (for문 활용하여 combination 구하기)

n, s = map(int, input().split())
A = list(map(int, input().split()))
saved = 0
for i in range(n):
    for j in range(i+1, n):
        for k in range(j+1, n):
            p = A[i] + A[j] + A[k]
            if p <= s and p > saved:
                saved = p
print(saved)

✔ for 문을 활용하여 combination을 구현하였다.

(2) 두번째 풀이 (itertools 🔧 활용)

from itertools import combinations

n, s = map(int, input().split())
cards = list(map(int, input().split()))
max_sum = 0

for c in combinations(cards, 3):
    if sum(c) <= s and sum(c) > max_sum:
        max_sum = sum(c)
print(max_sum)