내일배움캠프 51일차 TIL

코드카타

• 주사위 게임

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 함수를 작성해 주세요.

def solution(a, b, c, d):
    dice = [a, b, c, d]
    unique_numbers = set(dice)
    
    # 네 주사위 숫자가 모두 같은 경우
    if len(unique_numbers) == 1:
        return 1111 * dice[0]
    
    # 주사위 숫자들의 등장 횟수를 계산
    count_dict = {}
    for num in dice:
        if num in count_dict:
            count_dict[num] += 1
        else:
            count_dict[num] = 1
    
    # 세 주사위가 같은 숫자인 경우
    if 3 in count_dict.values():
        for key, value in count_dict.items():
            if value == 3:
                p = key
            else:
                q = key
        return (10 * p + q) ** 2
    
    # 두 개씩 같은 숫자인 경우
    if len(unique_numbers) == 2 and all(value == 2 for value in count_dict.values()):
        nums = list(unique_numbers)
        p, q = nums[0], nums[1]
        return (p + q) * abs(p - q)
    
    # 두 주사위가 같고 나머지 두 주사위가 서로 다른 경우
    if len(unique_numbers) == 3:
        for key, value in count_dict.items():
            if value == 2:
                p = key
            else:
                if 'q' not in locals():
                    q = key
                else:
                    r = key
        return q * r
    
    # 네 주사위 숫자가 모두 다른 경우
    if len(unique_numbers) == 4:
        return min(unique_numbers)