[Python] 프로그래머스 주사위 게임 3

문제 설명

1부터 6까지 숫자가 적힌 주사위가 네 개 있습니다. 네 주사위를 굴렸을 때 나온 숫자에 따라 다음과 같은 점수를 얻습니다.

• 네 주사위에서 나온 숫자가 모두 p로 같다면 1111 × p점을 얻습니다.
• 세 주사위에서 나온 숫자가 p로 같고 나머지 다른 주사위에서 나온 숫자가 q(p ≠ q)라면 (10 × p + q)2 점을 얻습니다.
• 주사위가 두 개씩 같은 값이 나오고, 나온 숫자를 각각 p, q(p ≠ q)라고 한다면 (p + q) × |p - q|점을 얻습니다.
• 어느 두 주사위에서 나온 숫자가 p로 같고 나머지 두 주사위에서 나온 숫자가 각각 p와 다른 q, r(q ≠ r)이라면 q × r점을 얻습니다.
• 네 주사위에 적힌 숫자가 모두 다르다면 나온 숫자 중 가장 작은 숫자 만큼의 점수를 얻습니다.

네 주사위를 굴렸을 때 나온 숫자가 정수 매개변수 a, b, c, d로 주어질 때, 얻는 점수를 return 하는 solution 함수를 작성해 주세요.

---

제한사항

• a, b, c, d는 1 이상 6 이하의 정수입니다.

---

주사위를 던지고, 각 숫자가 나오는 횟수를 알아야함

1) set 활용

2) dictionary 활용

- defaultdict 활용

--> 초기화 없이 바로 dict에 저장 가능 (그냥 dict 사용 시 각 값에 대해서 0으로 설정 해주고 카운트 해야됨)

우선 dict를 사용하면 각 dict에 몇 개의 숫자가 저장되는지에 따라 주사위 숫자(Key):나온 횟수(Value)로 저장됨

이때 Value를 기준으로 dict를 정렬하면 나온 주사위의 숫자들이 적게 나온 순서대로 정렬됨

 

이를 기준으로 조건에 맞게 계산하기

import math
from collections import defaultdict

def solution(a, b, c, d):
    dice_list = [a, b, c, d]
    dice_dict = defaultdict(int)
    for n in dice_list:
        dice_dict[n] += 1
    sort_dict = sorted(dice_dict, key = lambda x:dice_dict[x])
    
    if len(dice_dict) == 4:
        return min(dice_list)
    elif len(dice_dict) == 3:
        return sort_dict[0] * sort_dict[1]
    elif len(dice_dict) == 2:
        if dice_dict[sort_dict[0]] == 2:
            return (sort_dict[0] + sort_dict[1]) * abs(sort_dict[0] - sort_dict[1])
        else:
            return (10 * sort_dict[1] + sort_dict[0]) ** 2
    else:
        return 1111 * sort_dict[0]

---

https://school.programmers.co.kr/learn/courses/30/lessons/181916

프로그래머스