[브루트포스] 백준(BOJ) 2798번 - 블랙잭 python

세 장의 카드를 고르는 모든 경우를 고려하는 문제

* 브루트 포스(brute force)
brute: 무식한, force: 힘 무식한 힘으로 해석할 수 있다.
완전탐색 알고리즘. 즉, 가능한 모든 경우의 수를 모두 탐색하면서 요구조건에 충족되는 결과만을 가져온다.
이 알고리즘의 강력한 점은 예외 없이 100%의 확률로 정답만을 출력한다.

문제 보기

문제

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

예제 입력 1 복사

5 21
5 6 7 8 9

예제 출력 1 복사

21

예제 입력 2 복사

10 500
93 181 245 214 315 36 185 138 216 295

예제 출력 2 복사

497

문제풀이

1. itertools 라이브러리 사용

* itertools
파이썬에서 반복되는 데이터를 처리하는 기능을 포함하고 있는 라이브러리
코딩테스트에서 유용하게 사용할 수 있는 클래스 permutations, combinations를 제공한다.
- permutations: iterable 객체에서 r개의 데이터를 뽑아 일렬로 나열하는 모든 경우(순열)을 계산해준다
- combinations: iterable 객체에서 r개의 데이터를 뽑아 순서를 고려하지 않고 나열하는 모든 경우(조합)을 계산해준다.

from itertools import combinations

N, M = map(int, input().split())
lst = list(map(int, input().split()))

sum_list = []

combi_list = list(combinations(lst, 3))
for cards in combi_list:
    if sum(cards) <= M: # M보다 작은 수만 sum list에 넣어준다.
        sum_list.append(sum(cards))

print(max(sum_list))

2. for문 사용

=> 3중 for문을 사용하기 때문에 가독성도 떨어지고 복잡하다.

N, M = map(int, input().split())
lst = list(map(int, input().split()))

biggest_sum = 0

for i in range(N):
    for j in range(i+1, N):
        for k in range(j+1, N):
            if lst[i]+lst[j]+lst[k] > M:
                continue
            else:
                biggest_sum = max(biggest_sum, lst[i]+lst[j]+lst[k])

print(biggest_sum)

 

=> 두 방법간의 시간 차이는 거의 없다.