Programmers Java 코테 - 주사위 게임2

Q: 1부터 6까지 숫자가 적힌 주사위가 세 개 있습니다. 세 주사위를 굴렸을 때 나온 숫자를 각각 a, b, c라고 했을 때 얻는 점수는 다음과 같습니다.

• 세 숫자가 모두 다르다면 a + b + c 점을 얻습니다.
• 세 숫자 중 어느 두 숫자는 같고 나머지 다른 숫자는 다르다면 (a + b + c) × (a2 + b2 + c2 )점을 얻습니다.
• 세 숫자가 모두 같다면 (a + b + c) × (a2 + b2 + c2 ) × (a3 + b3 + c3 )점을 얻습니다.

세 정수 a, b, c가 매개변수로 주어질 때, 얻는 점수를 return 하는 solution 함수를 작성해 주세요.

---

딱히 큰 이유는 없지만 Stream를 써서 풀고싶어서 적용시켜본 답안.

import java.util.Arrays;

class Solution {
    public int solution(int a, int b, int c) {
        int[] intArray = {a, b, c};
        
        if(a == b && b == c) {
            return powNumber(intArray, 3);
        } else if(a == b || a == c || b == c) {
            return powNumber(intArray, 2);
        } else {
            return powNumber(intArray, 1);
        }
    }
    
    private int powNumber(int[] intArray, int pow) {
        int result = Arrays.stream(intArray)
            .map(num -> (int)Math.pow(num, pow))
            .sum();
        return  result = pow - 1 > 0 ? result * powNumber(intArray, pow-1) : result;
    }
    
}