[프로그래머스 - 자바] 귤 고르기

이번 주의 문제

경화는 과수원에서 귤을 수확했습니다. 경화는 수확한 귤 중 '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

생각해본 풀이

import java.util.*;
import java.io.*;

class Solution {
    public int solution(int k, int[] tangerine) {
        
        Map<Integer, Integer> map = new HashMap<>();
        
        int len = tangerine.length;
        
        for(int i = 0; i < len; i++) {
            map.put(tangerine[i], map.getOrDefault(tangerine[i], 0) + 1);
        }
        
        List<Integer> list = new ArrayList<>(map.values());
        list.sort(Collections.reverseOrder());
        
        
        int cnt = 0;
        
        for(int i = 0; i < list.size(); i++) {
            k -= list.get(i);
            cnt ++;
            
            if(k <= 0) break;
        }

        return cnt;
    }
}

함께 보면 좋은 자료구조

Map

Key와 value로 이루어진 자료형이다. 흔히 List 형태처럼 index로 값을 가져오는 것이 아니라 순서대로 저장되지 않는다. key를 통해 value를 가져 온다. key는 중복될 수 없으며, value는 중복될 수 있다.

(https://blog-of-gon.tistory.com/187#google_vignette)

• HashMap
  key와 value의 쌍으로만 구성이 될뿐 자료구조 안에 묶인 쌍들에 대한 순서는 보장할 수 없다.
  즉, 사용자는 키와 값이 구성되는 위치를 결정 하거나 알 수 없다.

• TreeMap
  key의 값을 이용해 순서대로 정렬하여 데이터를 저장하는 자료구조
  key값을 통한 탐색 뿐 아니라 key값의 정렬을 통한 탐색 등을 하기에 용의 하다.

• LinkedHashMap
  데이터를 입력한 순서대로 쌓아지며 데이터를 저장하는 자료구조
  배열, 리스트처럼 인덱싱 접근을 하기에 용의 하다.