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

문제

 

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

프로그래머스

---

풀이

def solution(a, b, c):
    answer = 0
    if a == b and b == c:
        answer = (a + b + c) * (a * a + b * b + c * c) * (a ** 3 + b ** 3 + c ** 3)
    elif a != b and b != c and c != a:
        answer = a + b + c
    else : 
        answer = (a + b + c) * (a * a + b * b + c * c)
    return answer

1. 세 글자가 모두 같다면 (a + b + c) * (a * a + b * b + c * c) * (a ** 3 + b ** 3 + c ** 3) 을 answer에 저장

2. 세 글자가 모두 다르다면 a + b + c 을 answer에 저장

3. 그 이외의 경우는 두 글자는 같고 한 글자만 다른 경우이므로 (a + b + c) * (a * a + b * b + c * c) 을 answer에 저장

4. answer을 반환