프로그래머스(자바, Java) - Level2. 귤 고르기

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

프로그래머스

---

주요 포인트 및 문제 리뷰

• n이라는 숫자가 m번 나오는 구조는 map을 사용하면 쉽게 만들 수 있었다.
• 문제의 요구 조건을 해결하기 위해 map의 value값이 큰 순서대로 k에서 값을 빼줘야 한다. 이때 내림차순 정렬을 하기 위해 Comparator라는 인터페이스 안에 compare 메서드를 람다식으로 사용한다.
• 원하는 조건을 설정하고 대소 비교를 통해 해당하는 숫자가 왼쪽으로(내림차순) 또는 오른쪽으로 올지(오름차순) 설정하면 된다.
• map의 key 값을 value 기준으로 정렬할 수 있다면 쉽게 풀 수 있었을 것 같다.

---

소스코드

import java.util.*;
class Solution {
    public int solution(int k, int[] tangerine) {
        int answer = 0;
        Map<Integer, Integer> map = new HashMap<>();

        for (Integer i : tangerine) {
            if (map.containsKey(i)) {
                map.put(i, map.get(i) + 1);
            } else {
                map.put(i, 1);
            }
        }

        ArrayList<Integer> arrayList = new ArrayList<>(map.keySet());

        Collections.sort(arrayList, (o1, o2) -> Integer.compare(map.get(o2), map.get(o1)));

        for (Integer num : arrayList) {
            int val = map.get(num);

            if (val >= k) {
                answer++;
                return answer;
            } else {
                k = k - val;
                answer++;
            }
        }

        return answer;
    }
}