많은 조건 분기, 구현
근우는 오늘 재미있는 카드 게임을 배우고 있다. 카드는 빨간색, 파란색, 노란색, 녹색의 네 가지 색이 있고, 색깔별로 1부터 9까지 숫자가 쓰여진 카드가 9장씩 있다. 카드는 모두 36(=4x9)장이다. 근우가 배운 카드 게임은 36장의 카드에서 5장을 뽑고, 아래와 같은 규칙으로 정수를 계산하는 것이다.
각 카드는 다음과 같이 나타낸다. 카드의 색깔은 영어 대문자 R, B, Y, G로 나타내는데, R은 빨간색, B는 파란색, Y는 노란색, G는 녹색을 뜻한다. 예를 들어서 Y8은 노란색 8을 나타내고, B5는 파란색 5를 나타낸다.
<점수를 정하는 규칙>
- 카드 5장이 모두 같은 색이면서 숫자가 연속적일 때, 점수는 가장 높은 숫자에 900을 더한다. 예를 들어, 카드가 Y4, Y3, Y2, Y5, Y6 일 때 점수는 906(=6+900)점이다.
- 카드 5장 중 4장의 숫자가 같을 때 점수는 같은 숫자에 800을 더한다. 예를 들어, 카드가 B3, R3, B7, Y3, G3 일 때 점수는 803(=3+800)점이다.
- 카드 5장 중 3장의 숫자가 같고 나머지 2장도 숫자가 같을 때 점수는 3장이 같은 숫자에 10을 곱하고 2장이 같은 숫자를 더한 다음 700을 더한다. 예를 들어, 카드가 R5, Y5, G7, B5, Y7 일 때 점수는 757(=5x10+7+700)점이다.
- 5장의 카드 색깔이 모두 같을 때 점수는 가장 높은 숫자에 600을 더한다. 예를 들어, 카드가 Y3, Y4, Y8, Y6, Y7 일 때 점수는 608(=8+600)점이다.
- 카드 5장의 숫자가 연속적일 때 점수는 가장 높은 숫자에 500을 더한다. 예를 들어 R7, R8, G9, Y6, B5 일 때 점수는 509(=9+500)점이다.
- 카드 5장 중 3장의 숫자가 같을 때 점수는 같은 숫자에 400을 더한다. 예를 들어 R7, Y7, R2, G7, R5 일 때 점수는 407(=7+400)점이다.
- 카드 5장 중 2장의 숫자가 같고 또 다른 2장의 숫자가 같을 때 점수는 같은 숫자 중 큰 숫자에 10을 곱하고 같은 숫자 중 작은 숫자를 더한 다음 300을 더한다. 예를 들어, R5, Y5, Y4, G9, B4 일 때 점수는 354(=5X10+4+300)점이다.
- 카드 5장 중 2장의 숫자가 같을 때 점수는 같은 숫자에 200을 더한다. 예를 들어, R5, Y2, B5, B3, G4 일 때 점수는 205(=5+200)점이다.
- 위의 어떤 경우에도 해당하지 않을 때 점수는 가장 큰 숫자에 100을 더한다. 예를 들어, R1, R2, B4, B8, Y5 일 때 점수는 108(=8+100)점이다.
입력으로 카드 5장이 주어질 때, 카드 게임의 점수를 구하는 프로그램을 작성하시오. 두 가지 이상의 규칙을 적용할 수 있는 경우에는 가장 높은 점수가 카드 게임의 점수이다.
첫째 줄부터 다섯째 줄까지 한 줄에 카드 하나씩 입력된다. 카드의 색깔과 숫자 사이에는 빈 칸이 하나 있다.
한 줄에 카드의 점수를 출력한다.
코드
# 첫 번째 리스트는 색깔의 개수를 저장
# 두 번째 리스트는 숫자의 개수를 저장 (0~9까지, 인덱스 0은 사용되지 않음)
# 세 번째 리스트는 입력된 숫자를 저장 (스트레이트 계산에 활용)
card = [[0]*4, [0]*10, []]
# 5개의 카드 입력을 처리
for _ in range(5):
color, idx = input().split()
# 색깔별로 개수 증가
if color == 'R':
card[0][0] += 1 # 빨간색 카운트 증가
elif color == 'B':
card[0][1] += 1 # 파란색 카운트 증가
elif color == 'Y':
card[0][2] += 1 # 노란색 카운트 증가
elif color == 'G':
card[0][3] += 1 # 초록색 카운트 증가
# 숫자별로 개수 증가
card[1][int(idx)] += 1
# 입력된 숫자 저장
card[2].append(int(idx))
# 같은 색깔 카드가 5장인 경우
if 5 in card[0]:
# 연속된 숫자인 경우 (스트레이트 플러시)
if max(card[2]) - min(card[2]) == 4:
print(max(card[2]) + 900)
# 연속된 숫자가 아닌 경우 (플러시)
else:
print(max(card[2]) + 600)
# 같은 숫자 카드가 4장인 경우 (포카드)
elif 4 in card[1]:
print(800 + card[1].index(4))
# 같은 숫자 카드가 3장인 경우
elif 3 in card[1]:
# 추가로 같은 숫자 카드가 2장인 경우 (풀하우스)
if 2 in card[1]:
print(card[1].index(3)*10 + card[1].index(2) + 700)
# 풀하우스가 아닌 경우 (트리플)
else:
print(card[1].index(3) + 400)
# 같은 숫자 카드가 2장인 경우
elif 2 in card[1]:
# 두 개의 페어인 경우 (투 페어)
if card[1].count(2) == 2:
nums = []
for i in range(1, 10):
if card[1][i] == 2:
nums.append(i)
print(min(nums) + max(nums)*10 + 300)
# 하나의 페어인 경우 (원 페어)
else:
print(card[1].index(2) + 200)
# 위의 모든 조건에 해당하지 않는 경우 (높은 카드)
else:
# 연속된 숫자인 경우 (스트레이트)
if max(card[2]) - min(card[2]) == 4:
print(max(card[2]) + 500)
# 그 외의 경우 (하이 카드)
else:
print(max(card[2]) + 100)
'알고리즘 > 백준' 카테고리의 다른 글
[Silver IV] 걷기 - 1459 (python) (0) | 2024.07.05 |
---|---|
[Silver IV] 유통기한 - 26083 (python) (0) | 2024.07.05 |
[Silver II] 한 줄로 서기 - 1138 (python) (0) | 2024.07.05 |
[Silver III] 우승자는 누구? - 5179 (python) (0) | 2024.07.05 |
[Silver III] 선배님 밥 사주세요! - 31869 (python) (0) | 2024.07.05 |