[프로그래머스] 주사위 게임 3 | 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점을 얻습니다.
• 네 주사위에 적힌 숫자가 모두 다르다면 나온 숫자 중 가장 작은 숫자 만큼의 점수를 얻습니다.
• 네 주사위를 굴렸을 때 나온 숫자가 정수 매개변수 a, b, c, d로 주어질 때, 얻는 점수를 return 하는 solution 함수를 작성해 주세요.

 

 

제한사항

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

입출력 예

 

입출력 예 설명

입출력 예 #1

• 예제 1번에서 네 주사위 숫자가 모두 2로 같으므로 1111 × 2 = 2222점을 얻습니다. 따라서 2222를 return 합니다.

입출력 예 #2

• 예제 2번에서 세 주사위에서 나온 숫자가 4로 같고 나머지 다른 주사위에서 나온 숫자가 1이므로 (10 × 4 + 1)2 = 412 = 1681점을 얻습니다. 따라서 1681을 return 합니다.

입출력 예 #3

• 예제 3번에서 a, d는 6으로, b, c는 3으로 각각 같으므로 (6 + 3) × |6 - 3| = 9 × 3 = 27점을 얻습니다. 따라서 27을 return 합니다.

입출력 예 #4

• 예제 4번에서 두 주사위에서 2가 나오고 나머지 다른 두 주사위에서 각각 5, 6이 나왔으므로 5 × 6 = 30점을 얻습니다. 따라서 30을 return 합니다.

입출력 예 #5

• 예제 5번에서 네 주사위 숫자가 모두 다르고 나온 숫자 중 가장 작은 숫자가 2이므로 2점을 얻습니다. 따라서 2를 return 합니다.

 

 

정답 코드

728x90

def solution(a, b, c, d):
    nums = {a, b, c, d}
    if len(nums) == 4:
        return min(nums)
    
    count = {}
    for num in (a, b, c, d):
        count[num] = count.get(num, 0) + 1
    
    if len(nums) == 1:
        return nums.pop() * 1111
    elif len(nums) == 2:
        if 3 in count.values():
            p = [k for k, v in count.items() if v == 3][0]
            q = [k for k in nums if k != p][0]
            return (10 * p + q) ** 2
        else:
            p, q = count.keys()
            return (p + q) * abs(p - q)
    else:
        p = [k for k, v in count.items() if v == 2][0]
        q, r = [k for k in nums if k != p]
        return q * r

 

풀이

count는 각 숫자가 주사위에서 등장한 횟수를 저장하는 딕셔너리 count.get(num, 0)은 딕셔너리에서 num에 해당하는 키가 있으면 그 값을 반환, 없으면 0을 반환

nums를 set 자료구조로 두어서 중복을 제거한다.
ex, {1, 2, 2, 3} ==> {1, 2, 3}

• len(nums) == 1:

모두 같은 숫자이기 때문에 return nums.pop() * 1111

• len(nums) == 2:

count의 값이 3인 키가 있는지를 검사하여 존재하면 해당 키를 p로, 그렇지 않으면 count의 키를 리스트로 가져온 후 첫 번째와 두 번째 키를 p와 q로 저장

• len(nums) ==3:

count의 값이 2인 키를 p로, 그 외의 두 숫자를 q와 r로 저장

• len(nums)==4:

가장 작은 수를 반환 ==> return min(nums)