[프로그래머스] 주사위 게임 3 - JAVA (자바)

문제 출처 : https://school.programmers.co.kr/learn/courses/30/lessons/181916

프로그래머스

 

• 문제 : 주사위 게임 3
  
  
• 문제 설명 :
  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 함수를 작성해 주세요.
  
  
• 풀이 :

import java.util.*;
import java.util.function.Function;
import java.util.stream.Collectors;

class Solution {
    public int solution(int a, int b, int c, int d) {
        List<Integer> arrInt = new ArrayList<>(Arrays.asList(a,b,c,d));

        Map<Integer, Integer> map = arrInt.stream().collect(Collectors.toMap(Function.identity(), f -> 1, Integer::sum));
        int[] keyList = map.keySet().stream().mapToInt(v -> v).toArray();
        int[] valueList = map.values().stream().mapToInt(v -> v).toArray();
        
        
        if(map.size() == 1) { // 4개중복
            return 1111 * a;
        } else if(map.size() == 2) {    // 2개 중복 (3:1, 2:2)
            if(valueList[0] == valueList[1]) {   // 2:2
                int [] mapKey = map.keySet().stream().mapToInt(v -> v).toArray();
                
                return (keyList[0] + keyList[1]) * Math.abs(keyList[0] - keyList[1]);
            } else {                            // 3:1
                if(valueList[0] > valueList[1]) {
                    return ((10 * keyList[0]) + keyList[1]) * ((10 * keyList[0]) + keyList[1]);
                } else {
                    return ((10 * keyList[1]) + keyList[0]) * ((10 * keyList[1]) + keyList[0]);
                }
            }
        } else if(map.size() == 3) {    // 2중복 2미중복
            int q = 0;
            int r = 0;
            
            if(valueList[0] == 2) {
                q = keyList[1];
                r = keyList[2];
            } else if (valueList[1] == 2) {
                q = keyList[0];
                r = keyList[2];
            } else {
                q = keyList[0];
                r = keyList[1];
            }
            
            return q * r;
        } else {
            return arrInt.stream().mapToInt(v -> v).min().getAsInt();
        }
    }
}