프로그래머스 : 주사위 게임 2 (파이썬)

문제

내답안

import math #math를 임포트하면 math 관련 함수를 쓸수있음.
#math.pow는 제곱의 값을 추출할수 있음.

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 #최종반환

 

결과

후기

math 라는 수학함수를 참조하여

제곱의 값을 산출함으로써 코드 생산성에 용이한것을 알게됨.