[프로그래머스] 귤🍊 고르기

문제

경화는 과수원에서 귤을 수확했습니다. 경화는 수확한 귤 중 'k'개를 골라 상자 하나에 담아 판매하려고 합니다. 그런데 수확한 귤의 크기가 일정하지 않아 보기에 좋지 않다고 생각한 경화는 귤을 크기별로 분류했을 때 서로 다른 종류의 수를 최소화하고 싶습니다.

예를 들어, 경화가 수확한 귤 8개의 크기가 [1, 3, 2, 5, 4, 5, 2, 3] 이라고 합시다. 경화가 귤 6개를 판매하고 싶다면, 크기가 1, 4인 귤을 제외한 여섯 개의 귤을 상자에 담으면, 귤의 크기의 종류가 2, 3, 5로 총 3가지가 되며 이때가 서로 다른 종류가 최소일 때입니다.

경화가 한 상자에 담으려는 귤의 개수 k와 귤의 크기를 담은 배열 tangerine이 매개변수로 주어집니다. 경화가 귤 k개를 고를 때 크기가 서로 다른 종류의 수의 최솟값을 return 하도록 solution 함수를 작성해주세요.

제한사항
1 ≤ k ≤ tangerine의 길이 ≤ 100,000
1 ≤ tangerine의 원소 ≤ 10,000,000

입출력 예

k	tangerine	result
6	[1, 3, 2, 5, 4, 5, 2, 3]	3
4	[1, 3, 2, 5, 4, 5, 2, 3]	2
2	[1, 1, 1, 1, 2, 2, 2, 3]	1

문제 출처

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

생각

• 크기가 서로 다른 귤의 종류가 적기를 원하기 때문에 각 귤 사이즈 별 귤의 개수를 담은 map을 생성한다
• 주어진 tangerine 리스트를 map의 value의 역순으로 정렬한다
  -> 개수가 많은 사이즈의 귤이 먼저 나온다
• 앞에서부터 k개를 자른 뒤 그 중 귤 사이즈의 종류를 count한다

코드

import java.util.*;

class Solution {
    public int solution(int k, int[] tangerine) {

        Map<Integer,Integer> cntMap = new HashMap<>();
        
        for (int t: tangerine){
            cntMap.put(t, cntMap.getOrDefault(t,0) + 1);
        }

        Integer[] sortedTangerine = Arrays.stream(tangerine)
                .boxed()
                .sorted(Comparator.<Integer, Integer>comparing(cntMap::get)
                        .reversed().thenComparingInt(i -> i))
                .toArray(Integer[]::new);
        

        return (int) Arrays.stream(sortedTangerine)
                .limit(k)
                .distinct()
                .count();
    }
}

다른 사람의 풀이

import java.util.*;

class Solution {
    public int solution(int k, int[] tangerine) {
        int answer = 0;
        HashMap<Integer,Integer> map =new HashMap<>();

        for (int t : tangerine) {
            map.put(t, map.getOrDefault(t, 0) + 1);
        }

        List<Integer> list = new ArrayList<>(map.keySet());
        list.sort((o1, o2) -> map.get(o2)-map.get(o1));

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

        return answer;
    }
}

• 사이즈 별 귤의 개수를 담은 map을 사용한 점은 동일했으나, input자체를 cnt 역순으로 정렬하는 대신 key 값을 value를 기준으로 정렬하여 더 간단하게 해결한 풀이를 기록한다
• 람다식을 쓰는 것 보다는 확실히 for문이 가독성이 더 좋은 것 같다 🤨