[Python] 주사위 게임 2

나의 풀이는 다음과 같다...!

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

    if (a != b) and (b != c) and (a != c):
        answer = a + b + c
    elif (a == b) and (b == c) and (a == c):
        answer = (a + b + c) * (a**2 + b**2 + c**2) * (a**3 + b**3 + c**3)
    else:
        answer = (a + b + c) * (a**2 + b**2 + c**2)

    return answer

근데 다른 사람의 풀이에서 너무 놀랬다... 그래서 적는 거다..!

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)

와.. set[a, b, c] 집합 개념으로 접근해서 풀었던 점이 너무 인상적이었다...ㅠㅠ