Basic23 주사위 게임 2
문제: 주사위 게임 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
, b
, c
가 매개변수로 주어질 때, 얻는 점수를 return 하는 solution 함수를 작성해 주세요.
제한사항
a
,b
,c
는 1이상 6이하의 정수입니다.
입출력 예
a | b | c | result |
---|---|---|---|
2 | 6 | 1 | 9 |
5 | 3 | 3 | 473 |
4 | 4 | 4 | 110592 |
입출력 예 설명
입출력 예 #1
- 예제 1번에서 세 주사위 숫자가 모두 다르므로 2 + 6 + 1 = 9점을 얻습니다. 따라서 9를 return 합니다.
입출력 예 #2
- 예제 2번에서 두 주사위 숫자만 같으므로 (5 + 3 + 3) × (52 + 32 + 32 ) = 11 × 43 = 473점을 얻습니다. 따라서 473을 return 합니다.
입출력 예 #3
- 예제 3번에서 세 주사위 숫자가 모두 같으므로 (4 + 4 + 4) × (42 + 42 + 42 ) × (43 + 43 + 43 ) = 12 × 48 × 192 = 110,592점을 얻습니다. 따라서 110592를 return 합니다.
Solution
switch case
문을 사용하여 세 개의 숫자가 다 같은 경우, 다 다른 경우 , 나머지로 나누었습니다. 또한 n제곱은a*a*a
이런 방식으로도 가능하며,pow(a, 3)
함수를 사용할 수 있습니다.
1
2
3
4
5
6
7
8
9
10
11
12
import Foundation
func solution(_ a:Int, _ b:Int, _ c:Int) -> Int {
switch (true) {
case a == b && b == c :
return Int(27 * pow(Float(a), 6)) //(3*a)*(3*a*a)*(3*a*a*a)
case a != b && b != c && a != c :
return a + b + c
default:
return (a + b + c) * (a*a + b*b + c*c)
}
}