본문 바로가기

Algorithm & SQL/Baekjoon

[Algorithm] [Python] BOJ/백준 - 2798_블랙잭

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장의 합을 출력한다.

제출 코드


N, M = map(int, input().split())
arr = list(map(int, input().split()))
result = 0
for i in range(N):
   for j in range(i+1, N):
       for k in range(j+1, N):
           if arr[i] + arr[j] + arr[k] > M:
               continue
           else:
               result = max(result, arr[i]+arr[j]+arr[k])
print(result)

코드설명


해당 문제는 모든 경우의 수를 다 뒤져봐야 하는 완전 탐색 문제이다.

Line 1 : N, M을 입력받아 int로 변환한다.

Line 2 : 카드에 쓰여 있는 수 들을 list화 하여 저장한다

Line 3 : 문제 조건에 적합한 수를 저장하여 반환하기 위한 변수 result 선언

Line 4~10 : 3개의 카드를 뽑아야 하며 이에 대한 모든 경우의 수를 살펴보기 위해 3중 for문을 이용한다.

​ 첫 for문 에서는 0

N , 두번째 for문 에서는 1

N , 세번쨰 for문 에서는 2~N를 순회하면서

​ 모든 경우의 수를 확인하며 각각의 값을 더한뒤, 세 카드의 값이 M보다 클 경우는 continue

​ M보다 크지 않을 경우, 일단은 정답의 가능성이 존재하므로 result 변수에 저장한다.

이렇게 모든 경우를 반복하면 결국, result 변수에는 M의 값 또는 M에 가장 가까운 값이 저장될 것이다.

Line 11 : 이를 출력한다.

다른 사람의 풀이


from itertools import combinations

card_num, target_num = map(int, input().split())
card_list = list(map(int, input().split())
biggest_num = 0

for cards in combinations(card_list, 3):
   temp_sum = sum(cards)
   if biggest_sum < temp_sum <= target_sum:
       biggest_sum = temp_sum
print(biggest_sum)

파이썬에 제공하는 순열 조합 라이브러리 itertools 모듈의 combinations함수를 사용하셨다.

순열과 조합 - combinations, permutations


ex )

  • 1, 2, 3의 숫자가 적힌 카드가 있을 때 , 이 중 두장을 꺼는 경우의 수 -> 12, 13, 21, 23, 31, 32
  • 'A', 'B', 'C'로 만들 수 있는 경우의 수 -> 'ABC', 'ACB', 'BAC', 'BCA', 'CAB', 'CBA'

파이썬에서는 itertools.permutation를 이용하면, for문을 사용하지 않고도 순열을 구할 수 있다.

import itertools

pool = ['A', 'B', 'C']
print(list(map(''.join, itertools.permutations(pool))))    # 3개의 원소로 수열 만들기
print(list(map(''.join, itertools.permutations(pool, 2)))) # 2개의 원소로 수열 만들기

# ['ABC', 'ACB', 'BAC', 'BCA', 'CAB', 'CBA']
# ['AB', 'AC', 'BA', 'BC', 'CA', 'CB']