[구현] (2-9) Counter()를 이용해 list를 hash로!

2024. 6. 14. 21:06개발의 흔적/코딩테스트

문제: 주사위 게임

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 명이 주사위 게임에 참여하였을 때, 가장 많은 상금을 받은 사람의 상금을 출력하는 프로그램 을 작성하시오

<입력설명>
첫째 줄에는 참여하는 사람 수 N(2<=N<=1,000)이 주어지고 그 다음 줄부터 N개의 줄에 사람 들이 주사위를 던진 3개의 눈이 빈칸을 사이에 두고 각각 주어진다.

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

입력예제

3

3 3 6

2 2 2

6 2 5

출력예제

12000

 

위 문제의 핵심은,  같은눈3개 / 같은눈2개만 / 같은눈x 의 사건들이 서로 독립적이라는 것을 알아차리는 것이다.

즉, 위 3개의 case들은 절대로 동시에 일어 날 수 없다.

 

그러나 만약 주사위 3개의 눈이 [2, 2, 5] 일때, 앞의 두개가 모두 2임을 알았을때를 가정해보자.

이때 다음 수가 2가 될 수도 있고 다른 수가 될 수도 있다.(위 상황에서는 무조건 5)

 

즉, 각 case가 같은눈3개 / 같은눈2개만 / 같은눈x 인지 알기 위해서는 모든 눈을 다 탐색해야지 알 수 있다.

 

이러한 논리를 이용하면서 코드를 간결히 쓰기위해서, dictionary를 사용하고자 하였고 (ex. [2, 2, 5] -> {2: 2, 5: 1} )

이를 위해서 collections library의 'Counter'함수를 사용한다!

 

from collections import Counter
dictionary = Counter('List')
#[6, 6, 6] -> {6 : 3}
#'abcabc' -> {'a':2 , 'b':2 , 'c' : 2}

 

 

import sys
from collections import Counter
input = sys.stdin.readline

N = int(input())
case_list = [list(map(int, input().split())) for _ in range(N)]
max = 0
for case in case_list:
    comb = Counter(case)
    # comb = {1:0, 2:0, 3:0, 4:0, 5:0, 6:0}
    # for num in case:
    #     comb[num] = comb[num] + 1
    price = 0
    for i in range(1, 7):
        if comb[i] == 3:
            price = 10000 + i*1000
            break
        if comb[i] == 2:
            price = 1000 + i*100
            break
        if comb[i] == 1:
            price = 100*i
    max = price if price > max else max
print(max)