📖 문제
https://school.programmers.co.kr/learn/courses/30/lessons/138476
더보기
제한사항
문제 설명
경화는 과수원에서 귤을 수확했습니다. 경화는 수확한 귤 중 '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
🔑 풀이
[성공]
- 귤 크기에 따른 수확량을 hash map에 저장한다.
- 수확량이 많은 값부터 원하는 개수가 될 때까지 상자에 담는다.
import java.util.*;
public class Solution {
public static int solution(int k, int[] tangerine) {
int answer = 0;
// value: 귤 크기, key: 수확량
Map<Integer, Integer> hm = new HashMap<>();
for (int i : tangerine) {
if (hm.containsKey(i))
hm.replace(i, hm.get(i) + 1);
else
hm.put(i, 1);
}
// 수확량을 기준으로 오름차순 정렬
List<Integer> keySet = new ArrayList<>(hm.keySet());
keySet.sort((o1, o2) -> hm.get(o2) - hm.get(o1));
// 수확량이 많은 귤부터 상자에 넣는다
for (int key : keySet) {
if (k > 0) {
// 더 담아야 하는 경우
k -= hm.get(key);
answer++;
// 다 담은 경우
if (k <= 0)
break;
} else {
answer++;
break;
}
}
return answer;
}
}
한번만에풀었당
hashmap 정렬하는 법 맨날 생각 안 나서 찾아보게 됨....
'알고리즘' 카테고리의 다른 글
[프로그래머스 Lv.2] H-index.java (1) | 2023.05.02 |
---|---|
[프로그래머스 Lv.2] 괄호 회전하기.java (0) | 2023.05.02 |
[프로그래머스 Lv.2] 멀리뛰기.java (0) | 2023.05.02 |
[프로그래머스 Lv.2] N개의 최소공배수.java (0) | 2023.04.29 |
[프로그래머스 Lv.2] 점프와 순간 이동.java (0) | 2023.04.27 |