백준 2476번: 주사위 게임

https://www.acmicpc.net/problem/2476

문제

1에서부터 6까지의 눈을 가진 3개의 주사위를 던져서 다음과 같은 규칙에 따라 상금을 받는 게임이 있다.

1. 같은 눈이 3개가 나오면 10,000원+(같은 눈)×1,000원의 상금을 받게 된다. 
2. 같은 눈이 2개만 나오는 경우에는 1,000원+(같은 눈)×100원의 상금을 받게 된다. 
3. 모두 다른 눈이 나오는 경우에는 (그 중 가장 큰 눈)×100원의 상금을 받게 된다.  

예를 들어, 3개의 눈 3, 3, 6이 주어지면 상금은 1,000+3×100으로 계산되어 1,300원을 받게 된다. 또 3개의 눈이 2, 2, 2로 주어지면 10,000+2×1,000 으로 계산되어 12,000원을 받게 된다. 3개의 눈이 6, 2, 5로 주어지면 그 중 가장 큰 값이 6이므로 6×100으로 계산되어 600원을 상금으로 받게 된다.

N(2 ≤ N ≤ 1,000)명이 주사위 게임에 참여하였을 때, 가장 많은 상금을 받은 사람의 상금을 출력하는 프로그램을 작성하시오.

입력

첫째 줄에는 참여하는 사람 수 N이 주어지고 그 다음 줄부터 N개의 줄에 사람들이 주사위를 던진 3개의 눈이 빈칸을 사이에 두고 각각 주어진다. 

출력

첫째 줄에 가장 많은 상금을 받은 사람의 상금을 출력한다.

예제 입력 1 

3
3 3 6
2 2 2
6 2 5

예제 출력 1 

12000

 

제출 코드

total = int(input())

max_prize = 0

for i in range(total):
    dice_list = [int(x) for x in input().split(' ')]

    #(주사위 눈, 횟수) 튜플로 저장
    dice_cnt = [(i, dice_list.count(i)) for i in range(1, 7) if dice_list.count(i)>0]

    #모두 같은 눈인 경우
    if len(dice_cnt) == 1:
        x, *_ = dice_cnt[0]
        prize = 10000+1000*x
    
    #두 개만 같은 눈인 경우
    elif len(dice_cnt) == 2:
        for (x, y) in dice_cnt: 
            if y == 2: 
                prize = 1000+x*100

    #세 개 모두 다른 눈인 경우
    else:
        x, *_ = dice_cnt[2]
        prize = x*100
    
    #최대 상금 
    if max_prize < prize:
        max_prize = prize

    #초기화
    dice_list.clear()
    dice_cnt.clear()

print(max_prize)

 

코멘트 )  백준 2480번: 주사위 세개 와 거의 동일한 문제로 코드를 재활용함