[Python] 백준 2798 브론즈2 블랙잭

JS로 코테 준비는 어떻게 된 것이냐 물으신다면..

무기한 보류입니다. 파이썬 못 잃어👽👽

 

알고리즘 한참 안 하다가 다시 기초부터 차근차근해보려 합니당.

올해 꼭 플레 간다.🔥

난 진지해요

 

---

 

백준 2798 브론즈 2 블랙잭

2798번: 블랙잭

 

문제

---

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

 

풀이

이 문제를 읽고 제일 먼저 생각난 방법은 카드 3장으로 만들 수 있는 모든 조합을 구해서 합을 구하고, M을 넘지 않는 수 중에 최댓값을 업데이트해 주는 방식.

 

조합은 라이브러리를 사용해 쉽게 구할 수 있다!

from itertools import *

n, m = map(int, input().split())
card = list(map(int, input().split())) 
ans = 0 # 최댓값을 계속 업데이트 해줄 변수

cardlist = list(combinations(card, 3)) # 조합을 뽑아주는 함수

for i in cardlist: # 조합 리스트를 돌면서
    a = sum(i) # 합을 구하고
    if a <= m and a > ans: # M을 넘지 않으면서 ans보다 크다면
        ans = a # ans 값을 a로 업데이트 해준다

print(ans)

 

itertools 라이브러리

여기서 사용한 itertools 라이브러리를 사용해 순열이나 조합을 쉽게 구할 수 있다.

 

permutations() : 순서를 고려한 경우의 수를 추출해 준다. (순열)

# 2가지 수로 이뤄진 순열 뽑기

from itertools import permutations

arr = [1, 2, 3, 4]
for p in permutations(arr, 2):
    print(p)

 

combinations() : 경우의 수 추출해 준다. (조합)

# 3가지 수로 이뤄진 조합 뽑기

from itertools import combinations

arr = [1, 2, 3, 4]
for c in combinations(arr,3):
    print(c)

 

위의 풀이에선 from itertools import * 로 라이브러리를 불러왔는데, itertools 라이브러리에 있는 모든 함수를 불러온다는 뜻이다.