백준 - 블랙잭[Python]

블랙잭

문제
카지노에서 제일 인기 있는 게임 블랙잭의 규칙은 상당히 쉽다. 카드의 합이 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장의 합을 출력한다.

입력	출력
5 21
5 6 7 8 9	21
10 500
93 181 245 214 315 36 185 138 216 295	497

Solution 1: Permutations

• 순열을 모두 구해서 그 합이 m이랑 같으면 바로 return하고, m보다 합이 작으면 answer에 대입 한 후, 합이 현재 answer보다 크면 교체해줌.
• 조합보다 시간 더 걸림 -> 순서를 따지기 때문에 조합보다 많은 경우가 나옴.
• 시간 524ms.

from itertools import permutations

n,m = map(int,input().split())
card = list(map(int,input().split()))

def find(n,m,card):
  answer = 0
  for i in permutations(card,3):
    total = sum(i)
    if total == m:
      return total
    elif total>m+1:
      continue
    else:
      answer = max(answer,sum(i))

  return answer
  
print(find(n,m,card))

Solution 2: Combinations

• Combinations으로 하면 시간이 덜 걸린다.
• 시간 144ms.

from itertools import combinations

n,m = map(int,input().split())
card = list(map(int,input().split()))

def find(n,m,card):
  answer = 0
  for i in combinations(card,3):
    total = sum(i)
    if total == m:
      return total
    elif total>m+1:
      continue
    else:
      answer = max(answer,sum(i))

  return answer
  
print(find(n,m,card))

다른 분의 풀이

import itertools

n, m = map(int, input().split())
cards = list(map(int, input().split()))

c = itertools.combinations(cards, 3)
print(max([sum(combination) if sum(combination) <= m else 0 for combination in c]))