[Programmers] Lv.0 / 주사위 게임 2 / Java

문제

 

풀이

class Solution {
    public int solution(int a, int b, int c) {
        int answer = a + b + c;
        
        if(a == b && b == c) {
            answer = (a + b + c) * (a * a + b * b + c * c) * (a * a * a + b * b * b + c * c * c);
        } else if((a == b && b != c) || (a != b && b == c) || (a == c && c != b)) {
            answer = (a + b + c) * (a * a + b * b + c * c);
        }
        return answer;
    }
}

class Solution {
    public int solution(int a, int b, int c) {
        
        int answer = 1;
        int count = 1;
        
        if(a == b || a == c || b == c) {
            count++;
        }
        if(a == b && b == c) {
            count++;
        }
        for(int i = 1; i <= count; i++) {
            answer *= (pow(a, i) + pow(b, i) + pow(c, i));
        }
        return answer;
    }

    private int pow(int base, int exponent) {
        if(exponent == 0) {
            return 1;
        }
        return base * pow(base, exponent-1);
    }
}

 

출처

https://school.programmers.co.kr/learn/courses/30/lessons/181930