[프로그래머스] 주사위 게임 3 자바 java

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

프로그래머스

문제 설명
1부터 6까지 숫자가 적힌 주사위가 네 개 있습니다. 네 주사위를 굴렸을 때 나온 숫자에 따라 다음과 같은 점수를 얻습니다.

네 주사위에서 나온 숫자가 모두 p로 같다면 1111 × p점을 얻습니다.
세 주사위에서 나온 숫자가 p로 같고 나머지 다른 주사위에서 나온 숫자가 q(p ≠ q)라면 (10 × p + q) 2 점을 얻습니다.
주사위가 두 개씩 같은 값이 나오고, 나온 숫자를 각각 p, q(p ≠ q)라고 한다면 (p + q) × |p - q|점을 얻습니다.
어느 두 주사위에서 나온 숫자가 p로 같고 나머지 두 주사위에서 나온 숫자가 각각 p와 다른 q, r(q ≠ r)이라면 q × r점을 얻습니다.
네 주사위에 적힌 숫자가 모두 다르다면 나온 숫자 중 가장 작은 숫자만큼의 점수를 얻습니다.
네 주사위를 굴렸을 때 나온 숫자가 정수 매개변수 a, b, c, d로 주어질 때, 얻는 점수를 return 하는 solution 함수를 작성해 주세요.

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

---

 

이 문제는 map을 사용하면 쉽게 풀 수 있는 문제이다. 

map에 a, b, c, d,를 추가하는데, 기존의 값이 없으면 1을 있다면 기존의 값+1 해서 나온 주사위의 수와 나온 횟수를 저장할 수 있도록 한다. 

map.put(a, map.getOrDefault(a,0)+1)을 각각 해주면 된다. 

 

그리고 이 문제에서 주어진 조건은 총 5개이다.

주사위로 나온 숫자가 

- 1개의 숫자만 나왔을 때   (1)

- 2개의 숫자가 나왔을 때 (2:2 혹은 1:3 두 가지 경우) 

- 3개의 숫자가 나왔을 때 (2:1:1)

- 4개의 숫자가 나왔을 때 (1:1:1:1)

 

 

1. 처음으로는 map을 초기화해준다. 

2. 2개의 수가 나왔을 때 사용하기 위해 리스트에 keySet을 초기화해 준다. 

3. switch문으로 map의 size별로 조건문을 실행하도록 구성해 준다. 

 

4-1. 주사위에서 나온 수의 개수가 1일 때는 1111*a를 해주면 된다. 

4-2. 주사위에서 나온 수의 개수가 2일 때는 그 비율이 1:3일 때와 2:2일 때를 if문으로 나누어 진행되도록 해주면 된다. 

미리 가지고 있던 keyset이 담긴 리스트를 가지고 빈도수를 따져 계산하도록 했다. 

4-3. 빈도수가 2인 수를 제외한 수 두 개를 곱해주면 되는데 간단하게 answer가 0이면 더해주고 아니면 곱해주도록 했다. 0이 아닌 이유는 이전 키가 더해져 있기 때문에 곱해주면 두 수의 곱이 계산될 수 있다. 

4-4. keySet 중 가장 작은 수를 반환하도록 했다. 

 

전체코드

import java.util.*;
class Solution {
    public int solution(int a, int b, int c, int d) {
        int answer = 0;
        Map<Integer , Integer> map = new HashMap<>();
        map.put(a,map.getOrDefault(a,0)+1);
        map.put(b,map.getOrDefault(b,0)+1);
        map.put(c,map.getOrDefault(c,0)+1);
        map.put(d,map.getOrDefault(d,0)+1);
        
        List<Integer> keys = new ArrayList<Integer>(map.keySet());
        // for(Integer v : keys) {
        //     System.out.println(v+" : "+ map.get(v) );
        // }
        
        switch(map.size()){
            case 1: answer = 1111*a; break;     //4
            case 2: 
                if(map.get(keys.get(0))==1||map.get(keys.get(1))==1){   //3:1
                    int p = map.get(keys.get(0)) == 3 ? keys.get(0) : keys.get(1);
                    int q = map.get(keys.get(0)) == 1 ? keys.get(0) : keys.get(1);
                    answer = (int)Math.pow((10 * p) + q, 2);
                }else{  //2:2
                    int p = keys.get(0);
                    int q = keys.get(1);
                    answer = (p+q)*Math.abs(p-q);
                }
                break;
            case 3: //2:1:1
                for(Integer val : map.keySet()){
                    if(map.get(val)!=2)
                        answer = answer==0? answer+val:answer*val;
                }
                break;
            case 4: //1:1:1:1
                int min = 7;
                for(Integer val : map.keySet()){
                    min = Math.min(val,min);
                }
                answer = min;
                break;
        }
        
        return answer;
    }
}