[프로그래머스] 주사위 게임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 함수를 작성해 주세요.

문제 풀이

---

문제 조건 정리

def solution(a, b, c, d):
    answer = 0
    return answer

기본 코드의 형태로 보면 a, b, c, d가 각각 네 주사위를 굴렸을 때 나온 숫자로 보인다.

• 조건 1, 네 주사위의 값이 p로 모두 같을 때, pppp

  $a=b=c=d\\Point = 1111 \times p$

• 조건 2. 세 주사위의 값이 p, 나머지 하나가 q, pppq

  $a=b=c\not=d\\ a=b=d\not=c\\ a=c=d\not=b\\ b=c=d\not=a\\ Point=(10 \times p + q)^2$

• 조건 3. 주사위가 두 개씩 같은 값이 나오고, 나온 숫자를 각각 p, q, ppqq

  $a=b\not=c=d\\ a=c\not=b=d\\ a=d\not=b=c\\ Point=(p+q) \times |p-q|$

• 조건 4. 어느 두 주사위에서 나온 숫자가 p로 같고 나머지 두 주사위에서 나온 숫자가 각각 p와 다른 q, r(q ≠ r), ppqr

  $a=b, \ a\not=c, \ a\not=d, \ c\not=d\\ a=c, \ a\not=b, \ a\not=d, \ b\not=d\\ a=d, \ a\not=b, \ a\not=c, \ b\not=c\\ b=c, \ b\not=a, \ b\not=d, \ a\not=d\\ b=d, \ b\not=a, \ b\not=c, \ a\not=d\\ c=d, \ c\not=a, \ c\not=b, \ a\not=b\\ Point = q \times r$

• 조건 5. 네 주사위에 적힌 숫자가 모두 다르다면, pqrs

  $a \not =b\\ a \not =c\\ a \not =d\\ b \not =c\\ b \not =d\\ c \not =d\\ Point = min(p, \ q, \ r, \ s)$

코드 작성

차근차근 조건문을 활용해 작성해보자.

def solution(a, b, c, d):
    # 주사위 수를 저장할 리스트
    dice = [a, b, c, d]
    
    # 조건 1, 네 주사위의 값이 `p`로 모두 같을 때, `pppp`
    if len(set(dice)) == 1:
        return 1111 * a
    
    # 조건 2. 세 주사위의 값이 `p`, 나머지 하나가 `q`, `pppq`
    elif a==b and b==c and c!=d: return (10 * a + d)**2
    elif a==b and b==d and d!=c: return (10 * a + c)**2
    elif a==c and c==d and d!=b: return (10 * a + b)**2
    elif b==c and c==d and d!=a: return (10 * b + a)**2

    # 조건 3. 주사위가 두 개씩 같은 값이 나오고, 나온 숫자를 각각 `p, q`, `ppqq`
    elif a==b and b!=c and c==d: return (a + d) * abs(a - d)
    elif a==c and c!=b and b==d: return (a + d) * abs(a - d)
    elif a==d and d!=b and b==c: return (a + c) * abs(a - c)

    # 조건 4. 어느 두 주사위에서 나온 숫자가 `p`로 같고 나머지 두 주사위에서 나온 숫자가 각각 `p`와 다른 `q, r(q ≠ r)`, `ppqr`
    elif a==b and a!=c and a!=d and c!=d: return c * d
    elif a==c and a!=b and a!=d and b!=d: return b * d
    elif a==d and a!=b and a!=c and b!=c: return b * c
    elif b==c and b!=a and b!=d and a!=d: return a * d
    elif b==d and b!=a and b!=c and a!=d: return a * c
    elif c==d and c!=a and c!=b and a!=b: return a * b

    # 조건 5. 네 주사위에 적힌 숫자가 모두 다르다면, `pqrs`
    if len(set(dice)) == 4:
        return min(dice)

뭔가 복잡한 조건문일수록 의외로 예외처리만 잘 한다면 쉽게 풀리는 경우가 많은데, 이 문제의 경우는 예외처리를 하다가 하다가 잘 안 되는 바람에 모든 조건을 천천히 처음부터 써내려갔다. 조건 1번과 조건 5번의 경우는 set을 이용해 조금 더 간단하게 처리할 수 있었다.