[프로그래머스(Java)] 주사위 게임 2

 

Level. 0

 

문제

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 함수를 작성해 주세요.

* 제한사항
- a, b, c는 1이상 6이하의 정수입니다.

 

풀이. 1

import java.util.Set;
import java.util.HashSet;

class Solution {
    public int solution(int a, int b, int c) {

        int answer = 0;
        Set<Integer> set = new HashSet<>();
        set.add(a); set.add(b); set.add(c);
        
        answer += a + b + c;
        if(set.size() <= 2)
            answer *= (a*a + b*b + c*c);
        if(set.size() <= 1)
            answer *= (a*a*a + b*b*b + c*c*c);
        
        return answer;

    }
   
}

 

중복된 숫자를 없애기 위해 set 자료구조를 사용해 보았다.

그러나 비교할 데이터가 3개밖에 되지 않기 때문에 조건문으로도 간단하게 풀 수 있었을 것 같다. 

 

 

---

다른 풀이

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

 

가독성이 좋은 풀이를 찾았다.

조건문에 맞게 count 해 주고, pow라는 함수를 통해 거듭재곱수를 구할 수 있도록 했다. 

 

 

 

 

---

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

프로그래머스