Lv.2 : 귤 고르기

👉 문제링크

프로그래머스

---

---

🔸 문제 분석 🔸

• k개의 귤을 선택할 때, 귤의 크기 종류의 최소값을 반환한다.

🔸 문제 풀이 🔸

• 귤의 크기 종류와 각각의 귤 개수를 카운트한다.
• 그리디하게 접근해서, 귤 개수가 많은 종류부터 k개를 고른다면 최소한의 종류로 선택할 수 있다.

---

🔸 코드 🔸

import java.util.HashMap;
import java.util.Map;
import java.util.ArrayList;
import java.util.List;

class Solution {
    public int solution(int k, int[] tangerine) {
        Map<Integer, Integer> map = new HashMap<>();
        
        for(int i : tangerine) {
            if(map.containsKey(i)) {
                map.put(i, map.get(i) + 1);
            } else {
                map.put(i, 1);
            }
        }
        
        List<Integer> keySet = new ArrayList<>(map.keySet());

        // Value 값으로 내림차순 정렬
        keySet.sort((o1, o2) -> map.get(o2).compareTo(map.get(o1)));

        int answer = 0;
        
        for (Integer key : keySet) {
            answer++;
            
            if(map.get(key) <= k) k -= map.get(key);
            else k = 0;
            
            if(k == 0)
                break;
        }
        
        return answer;
    }
}

🔸 코드 해석 🔸

• 귤 종류와 그 숫자를 map에 저장한다.
• 귤 개수로 내림차순 정렬한다.
• 귤 개수가 많은 종류부터 k개를 세고, 선택한 귤의 종류 수를 반환한다.

---

🔸 end 🔸

• 정렬문을 직접 작성해 넣는 방법이 오랜만이라서 포스팅을 찾아보았다.
  • List.sort()안에 직접 Comparator를 오버라이드해서 넣는방법과 람다식으로 넣는 두 가지 방법을 잊지 말자.
• 다른 풀이 방법을 찾아봤는데, 다 비슷한 것 같다.