[Level.0] 주사위 게임(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 이하의 정수입니다.

문제 해결

def solution(a, b, c, d):
    array = sorted([a, b, c, d])
    dicti = {}
    for arr in array:
        dicti[arr] = array.count(arr)
    keys = sorted(dicti.keys(), key=lambda x: dicti[x], reverse=True) # 빈도수가 높은 순서대로 키를 정렬
    if len(keys) == 1:
        return 1111 * keys[0]
    elif len(keys) == 2:
        if dicti[keys[0]] == 3: # keys[0] == 3이라는 것은 keys[1] == 1임
            return (10 * keys[0] + keys[1])**2
        else: # keys[0] != 3이 아니라면 이는 keys[0] == 2라는 의미
            return (keys[0] + keys[1]) * abs(keys[0] - keys[1])
    elif len(keys) == 3:
        return keys[1] * keys[2]
    else:
        return array[0]