[Programmers] #181930 - 주사위 게임 2 [Java][C++][Python]
세 주사위 눈의 일치 개수에 따라 분기 공식으로 점수를 계산하는 워밍업 문제.
[Programmers] #181930 - 주사위 게임 2 [Java][C++][Python]
1. 아이디어
세 주사위 눈 a, b, c가 모두 같은지, 그중 두 개만 같은지, 모두 다른지를 조건문으로 구분해서 각 경우에 맞는 점수 공식을 적용한다. 모두 같으면 $(a+b+c) \times (a^2+b^2+c^2) \times (a^3+b^3+c^3)$, 두 개만 같으면 $(a+b+c) \times (a^2+b^2+c^2)$, 모두 다르면 $a+b+c$를 반환한다.
2. 복잡도
| 접근 | 시간 | 공간 |
|---|---|---|
| 풀이 | $O(1)$ | $O(1)$ |
3. 코드
풀이 [Java][C++][Python]
1
2
3
4
5
6
7
8
9
10
11
class Solution {
public int solution(int a, int b, int c) {
if (a == b && b == c) {
return (a + b + c) * (a * a + b * b + c * c) * (a * a * a + b * b * b + c * c * c);
} else if (a == b || b == c || c == a) {
return (a + b + c) * (a * a + b * b + c * c);
} else {
return a + b + c;
}
}
}
1
2
3
4
5
6
7
8
9
10
11
12
#include <bits/stdc++.h>
using namespace std;
int solution(int a, int b, int c) {
if (a == b && b == c) {
return (a + b + c) * (a * a + b * b + c * c) * (a * a * a + b * b * b + c * c * c);
} else if (a == b || b == c || c == a) {
return (a + b + c) * (a * a + b * b + c * c);
} else {
return a + b + c;
}
}
1
2
3
4
5
6
7
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 c == a:
return (a + b + c) * (a**2 + b**2 + c**2)
else:
return a + b + c
This post is licensed under CC BY 4.0 by the author.