본문 바로가기

공부/알고리즘

[백준 2798/완전탐색] 블랙잭 - Python3

문제

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

# 카드의 갯수와, 넘지 말아야 할 값을 입력받는다. 
card_num, value = map(int, input().split())

# 카드의 값들을 입력받는다.
cards = list(map(int,input().split(' ')))

# 값을 저장하기 위한 answer 선언
answer = 0

# i, j, k 3개의 인덱스를 이용해서 3중 for문을 돌린다. 
# (1, 2, 3)을 1번 카드, 2번카드, 3번 카드의 합이라 할 때 
# 아래 for문은 (1, 2, 3) 부터 (3, 4, 5)까지 돌게 된다. 
for i in range(card_num - 2):
    for j in range(i+1, card_num - 1):
        for k in range(j+1, card_num):
            temp = cards[i] + cards[j] + cards[k]
            # 지금 경우에 구한 합이 value를 넘거나
            if temp > value:
                continue;
            
            # 이전에 구한 값보다 작다면 다른 행동을 취하지 않는다.
            elif temp < answer:
                continue;
                
            # value보다 작고, 이전 결과보다 가장 큰 값이 answer에 저장된다. 
            else:
                answer = temp
print(answer)

설명

  1. 카드의 갯수, 넘지 말아야 할 값, 각 카드의 값들을 입력받는다.
  2. 3중 for문을 순회하면서 (1, 2, 3) 부터 (3, 4, 5)까지 순회하면서 가장 큰 값을 저장한다.
  3. 3개의 카드를 더한 값이 value를 넘거나, 이전 answer보다 작다면 버린다. 
  4. answer를 print한다.

후기

전형적인 브루트 포스(가능한 모든 경우를 탐색하는) 방법으로, 쉽게 구현했다. 사실 이 방법으로 구현해도 되는 이유는 N(카드의 갯수)이 100 이하라고 주어졌기 때문이다. 지금 알고리즘의 시간복잡도는 O(N^3)로, 다른 문제도 이렇게 풀려고 했다가는 효율성 테스트에서 럼블 궁을 볼 가능성이 매우 크다. 따라서, 더 효율적으로 짜는 방법 연구가 필요하다~