[Java] 주사위 게임 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이하의 정수입니다.

 

import java.util.HashSet;

class Solution {
    public int solution(int a, int b, int c) {
        int answer = 0;        

        HashSet<Integer> diceSet = new HashSet<Integer>();
        diceSet.add(a);
        diceSet.add(b);
        diceSet.add(c);
        
        if(diceSet.size() == 3) {
            answer = a + b + c;
        }
        else if(diceSet.size() == 2) {
            answer = (a + b + c) * (int)(Math.pow(a, 2) + Math.pow(b, 2) + Math.pow(c, 2));
        }
        else if(diceSet.size() == 1) {
            answer = (a + b + c) * 
                (int)(Math.pow(a, 2) + Math.pow(b, 2) + Math.pow(c, 2)) *
                (int)(Math.pow(a, 3) + Math.pow(b, 3) + Math.pow(c, 3));
        }
        
        return answer;
    }
}

각 숫자의 중복을 알기위해 hashSet 을 썼는데 사실 그냥 비교연산으로 해결해도 될 문제였다.

 

 

출처

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