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

[unrated] 주사위 게임 2 - 181930

문제 링크

성능 요약

메모리: 10.2 MB, 시간: 0.00 ms

구분

코딩테스트 연습 > 코딩 기초 트레이닝

채점결과

Empty

문제 설명

1부터 6까지 숫자가 적힌 주사위가 세 개 있습니다. 세 주사위를 굴렸을 때 나온 숫자를 각각 a, b, c라고 했을 때 얻는 점수는 다음과 같습니다.

• 세 숫자가 모두 다르다면 a + b + c 점을 얻습니다.
• 세 숫자 중 어느 두 숫자는 같고 나머지 다른 숫자는 다르다면 (a + b + c) × (a² + b² + c² )점을 얻습니다.
• 세 숫자가 모두 같다면 (a + b + c) × (a² + b² + c² ) × (a³ + b³ + c³ )점을 얻습니다.

세 정수 a, b, c가 매개변수로 주어질 때, 얻는 점수를 return 하는 solution 함수를 작성해 주세요.

---

제한사항

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

---

입출력 예

a	b	c	result
2	6	1	9
5	3	3	473
4	4	4	110592

---

입출력 예 설명

입출력 예 #1

• 예제 1번에서 세 주사위 숫자가 모두 다르므로 2 + 6 + 1 = 9점을 얻습니다. 따라서 9를 return 합니다.

입출력 예 #2

• 예제 2번에서 두 주사위 숫자만 같으므로 (5 + 3 + 3) × (5² + 3² + 3² ) = 11 × 43 = 473점을 얻습니다. 따라서 473을 return 합니다.

입출력 예 #3

• 예제 3번에서 세 주사위 숫자가 모두 같으므로 (4 + 4 + 4) × (4² + 4² + 4² ) × (4³ + 4³ + 4³ ) = 12 × 48 × 192 = 110,592점을 얻습니다. 따라서 110592를 return 합니다.

정답

import math

def solution(a, b, c):
    answer = 0
    
    if a == b and b == c and a == c:
        answer = (a + b + c) * ((math.pow(a, 2)) + (math.pow(b, 2)) + (math.pow(c, 2))) * ((math.pow(a, 3)) + (math.pow(b, 3)) + (math.pow(c, 3)))
    elif a == b or b == c or a == c:
        answer = (a + b + c) * ((math.pow(a, 2)) + (math.pow(b, 2)) + (math.pow(c, 2)))
    else:
        answer = (a + b + c)
    
    return answer

출처: 프로그래머스 코딩 테스트 연습, https://programmers.co.kr/learn/challenges