728x90
반응형
# 주사위게임 2
# 1 - 6 까지 주사위가 세걔
# 각 a,b,c일 때 모두 다르면 a+b+c
# 두개만 같다면 (a+b+c)*(a^2+b^2+c^2)
# 모두 같다면(a+b+c)*(a^2+b^2+c^2)*(a^3+b^3+c^3)
# 제곱은 a**2 세제곱은 a**3으로 표현 가능하다는걸 깨달았다.
def solution(a, b, c):
sum=(a+b+c)
sum2=(a*a)+(b*b)+(c*c)
sum3=(a*a*a)+(b*b*b)+(c*c*c)
if a==b==c:
return int(sum*sum2*sum3)
elif (a!=b and b==c or a==c) or (a!=c and b==c or a==b) or (b !=c and b==a or c==a):
return int(sum*sum2)
elif a!=b and b!=c and a!=c:
return int(sum)
print(solution(2,6,1)) # 9
print(solution(5,3,3)) # 473
print(solution(4,4,4)) # 110592
728x90
반응형
'Algorithm > Python' 카테고리의 다른 글
[programmers]python 181832 정수 나선형으로 배치하기 (1) | 2023.12.19 |
---|---|
[programmers]python 92334 신고결과받기 (0) | 2023.12.16 |
[programmers]python 120956 옹알이(1) (0) | 2023.12.15 |
[programmers]python 181864 문자열 바꿔서 찾기 (1) | 2023.12.11 |
[programmes]python 181853 뒤에서5등까지 (0) | 2023.12.11 |