코테 Lv0 - 주사위 게임 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 함수를 작성해 주세요.

제한사항

• a, b, c, d는 1 이상 6 이하의 정수입니다.

입출력 예

입출력 예 #1

Input : 2(a), 2(b), 2(c), 2(d) / Output : 2222

예제 1번에서 네 주사위 숫자가 모두 2로 같으므로 1111 × 2 = 2222점을 얻습니다. 따라서 2222를 return 합니다.

입출력 예 #2

Input : 4(a), 1(b), 4(c), 4(d) / Output : 1681

예제 2번에서 세 주사위에서 나온 숫자가 4로 같고 나머지 다른 주사위에서 나온 숫자가 1이므로 (10 × 4 + 1)2 = 412 = 1681점을 얻습니다. 따라서 1681을 return 합니다.

입출력 예 #3

Input : 6(a), 3(b), 3(c), 6(d) / Output : 27

예제 3번에서 a, d는 6으로, b, c는 3으로 각각 같으므로 (6 + 3) × |6 - 3| = 9 × 3 = 27점을 얻습니다. 따라서 27을 return 합니다.

입출력 예 #4

Input : 2(a), 5(b), 2(c), 6(d) / Output : 30

예제 4번에서 두 주사위에서 2가 나오고 나머지 다른 두 주사위에서 각각 5, 6이 나왔으므로 5 × 6 = 30점을 얻습니다. 따라서 30

입출력 예 #5

Input : 6(a), 4(b), 2(c), 5(d) / Output : 2

예제 5번에서 네 주사위 숫자가 모두 다르고 나온 숫자 중 가장 작은 숫자가 2이므로 2점을 얻습니다. 따라서 2를 return 합니다.

제출코드

import java.util.Comparator;
import java.util.Iterator;
import java.util.Map;
import java.util.function.Function;
import java.util.stream.Collectors;
import java.util.stream.IntStream;

class Solution {
    public int solution(int a, int b, int c, int d) {
        Map<Integer, Integer> map = IntStream.of(a, b, c, d)
            .boxed()
            .collect(Collectors.toMap(Function.identity(), i -> 1, Integer::sum));
        Iterator<Integer> iterator = map.keySet().stream().sorted(Comparator.comparingInt(map::get).reversed()).iterator();

        if (map.size() == 1) {
            int p = iterator.next();
            return 1111 * p;
        } else if (map.size() == 2) {
            int p = iterator.next();
            int q = iterator.next();
            return map.containsValue(2) ?
                (p + q) * Math.abs(p - q) : (10 * p + q) * (10 * p + q);
        } else if (map.size() == 3) {
            int p = iterator.next();
            int q = iterator.next();
            int r = iterator.next();
            return q * r;
        } else {
            return map.keySet().stream().min(Comparator.naturalOrder()).orElse(0);
        }
    }
}

1. 주어진 4개의 숫자를 Stream화 하면 key는 주사위 숫자, value는 중복 개수를 가지는 Map으로 변환합니다.
2. 해당 Map을 중복 개수가 많은 순서로 정렬 후 Iterator화 합니다.
3. Map에 길이가 1개라면 네 주사위에서 나온 숫자가 모두 같으므로 1111 * p를 반환합니다. 
4. Map에 길이가 2개고 중복 개수 중 3이 존재한다면 (10 × p + q)*2(제곱)를 반환합니다.
5. Map에 길이가 2개고 중복 개수 중 2가 존재한다면 (p + q) × |p - q|를 반환합니다.
6. Map에 길이가 3개라면 q × r를 반환합니다.
7. Map에 길이가 4개라면 key값 중 가장 작은값을 반환합니다.

---

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

프로그래머스