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

def solution(a, b, c):
    if a == b == c :
        return (a+b+c)*(a**2+b**2+c**2)*(a**3+b**3+c**3)
    elif a == b or b == c or a == c:
        return (a+b+c)*(a**2+b**2+c**2)
    else:
        return a+b+c

def solution(a, b, c):
    check=len(set([a,b,c]))
    if check==1:
        return 3*a*3*(a**2)*3*(a**3)
    elif check==2:
        return (a+b+c)*(a**2+b**2+c**2)
    else:
        return (a+b+c)