[Python] Baekjoon 백준 2798번 블랙잭

[Python] Baekjoon 백준 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중 for문으로 계산

a, b = map(int, input().split())
num = list(map(int, input().split()))

su = 0
chk = False

for i in range(0, len(num) -2):
    for j in range(i + 1, len(num) -1):
        for k in range(j + 1, len(num)):
            if su < num[i] + num[j] + num[k] <= b:
                su = num[i] + num[j] + num[k]
            if su == b:
                chk = True
                break
        if chk == True:
            break
    if chk == True:
            break

print(su)

3중 for문 으로 3개의 수를 뽑고

su 보다 크고 입력된 값 b 보다 같거나 작은 수를 뽑아 su 에 저장

조금 번잡하게 보이지만 3개 더한값이 입력받은 b 와 같을경우 break

 

하지만

 

문제 의도가 이게 아닌듯하다

문제 알고리즘 분류 가 브루트포스 알고리즘 이다

 

그래서 combinations() 사용

from itertools import combinations

a, b = map(int, input().split())
num = list(map(int, input().split()))

su = 0

for s in combinations(num, 3):
    if su < sum(s) <= b:
        su = sum(s)

print(su)

 

combinations()는 리스트 내의 모든 조합을 나타낼수있다

combinations(리스트, 인자값)

 

위 3중 for문과 같은 조건으로 계산

 

 

추가

 

from itertools import permutations

하나의 리스트에서 모든 순열

 

from itertools import combinations

하나의 리스트에서 모든 조합

 

from itertools import product

두개 이상의 리스트에서 모든 조합

 

 

참고 사이트

 

itertools

https://docs.python.org/ko/3/library/itertools.html

 

itertools.combinations(iterable, r)

https://docs.python.org/ko/3/library/itertools.html#itertools.combinations

 

파이썬(Python) 리스트 모든 조합 구하기

https://ourcstory.tistory.com/414

 

itertools를 이용해 순열과 조합 구하기

https://itholic.github.io/python-combination-permutation/