[프로그래머스(Java)] 주사위 게임 3 / Collections.sort()

 

 

Level. 0

 

문제

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 이하의 정수입니다.

 

풀이

import java.util.Map;
import java.util.HashMap;
import java.util.Arrays;
import java.util.ArrayList;
import java.util.Collections;

class Solution {
    public int solution(int a, int b, int c, int d) {
        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);
        
        ArrayList<Integer> keys = new ArrayList<Integer>(map.keySet());
        Collections.sort(keys);
        if(map.size() == 1) return 1111 * keys.get(0);
        if(map.size() == 2) {
            if(map.get(keys.get(0)) == 2)
                return (keys.get(0) + keys.get(1)) * Math.abs(keys.get(0) - keys.get(1));
            if(map.get(keys.get(0)) == 1)
                return (10*keys.get(1) + keys.get(0)) * (10*keys.get(1) + keys.get(0));
            if(map.get(keys.get(1)) == 1)
                return (10*keys.get(0) + keys.get(1)) * (10*keys.get(0) + keys.get(1));
        }
        if(map.size() == 3){
            if(map.get(keys.get(0)) == 2)
                return keys.get(1)*keys.get(2);
            
            if(map.get(keys.get(1)) == 2)
                return keys.get(0)*keys.get(2);
            
            if(map.get(keys.get(2)) == 2)
                return keys.get(0)*keys.get(1);
        }
        
        return keys.get(0);
    }
}

 

해결방법

HashMap에는 등장하는 주사위숫자별 나온 횟수를 저장하였고,

ArrayList는 등장하는 주사위 숫자를 오름차순으로 정렬하여 저장하였다. 

저장한 데이터를 통해 아래와 같은 조건문을 돌려주었다.

1. 등장한 주사위의 숫자가 1가지 종류인 경우. 네 주사위에서 나온 숫자가 모두 같다는 의미이다. 
2. 등장한 주사위의 숫자가 2가지 종류인 경우. 주사위에서 나온 숫자가 2:2로 같거나, 3:1로 같다는 의미이다.
3. 등장한 주사위의 숫자가 3가지 종류인 경우. 주사위에서 나온 숫자가 2:1:1로 같다는 의미이다.
4. 위 조건문을 모두 통과했을 경우 주사위에서 나온 숫자가 모두 다르다는 의미이다. 

 

---

다른 풀이

import java.util.Arrays;

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

        int[] dice = { a, b, c, d };
        Arrays.sort(dice);

        int ans = 0;

        if (dice[0] == dice[3]) {
            ans = 1111 * dice[3];
        } else if (dice[0] == dice[2] || dice[1] == dice[3]) {
            ans = (int) Math.pow(dice[1] * 10 + (dice[0] + dice[3] - dice[1]), 2);
        } else if (dice[0] == dice[1] && dice[2] == dice[3]) {
            ans = (dice[0] + dice[3]) * (dice[3] - dice[0]);
        } else if (dice[0] == dice[1]) {
            ans = dice[2] * dice[3];
        } else if (dice[1] == dice[2]) {
            ans = dice[0] * dice[3];
        } else if (dice[2] == dice[3]) {
            ans = dice[0] * dice[1];
        } else {
            ans = dice[0];
        }

        return ans;
    }
}

---

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

프로그래머스