경화는 과수원에서 귤을 수확했습니다. 경화는 수확한 귤 중 'k'개를 골라 상자 하나에 담아 판매하려고 합니다. 그런데 수확한 귤의 크기가 일정하지 않아 보기에 좋지 않다고 생각한 경화는 귤을 크기별로 분류했을 때 서로 다른 종류의 수를 최소화하고 싶습니다.
예를 들어, 경화가 수확한 귤 8개의 크기가 [1, 3, 2, 5, 4, 5, 2, 3] 이라고 합시다. 경화가 귤 6개를 판매하고 싶다면, 크기가 1, 4인 귤을 제외한 여섯 개의 귤을 상자에 담으면, 귤의 크기의 종류가 2, 3, 5로 총 3가지가 되며 이때가 서로 다른 종류가 최소일 때입니다.
경화가 한 상자에 담으려는 귤의 개수 k
와 귤의 크기를 담은 배열 tangerine
이 매개변수로 주어집니다. 경화가 귤 k개를 고를 때 크기가 서로 다른 종류의 수의 최솟값을 return 하도록 solution 함수를 작성해주세요.
k
≤ tangerine
의 길이 ≤ 100,000tangerine
의 원소 ≤ 10,000,000k | 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 |
입출력 예 #1
입출력 예 #2
입출력 예 #3
문제를 잘 살펴보면 중복된 수가 가장 적은 숫자들을 최대한 피해가며 귤박스를 채워주면 되는 것을 알 수 있다.
각각의 수가 몇개씩 존재하는지는 Map
자료형에 담아주면 쉽게 파악할 수 있다. 이렇게 파악한 Map을 토대로 키에 해당하는 value
들을 다시 List
에 담아주고 이를 역순으로 정렬해준다.
ex) [1, 3, 2, 5, 4, 5, 2, 3] → [1=1, 2=2, 3=2, 4=1, 5=2
] → [2, 2, 2, 1, 1]
이렇게 정렬을 해주면 중복된 수가 가장 적은 숫자들을 제일 나중에 활용하게 된다.
위와 같은 규칙을 빨리 찾을 수 있었기에 문제를 해결할 수 있었다고 생각한다! 굿굿!
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, 1);
} else {
map.put(i, map.get(i) + 1);
}
}
List<Integer> list = new ArrayList<>(map.values());
list.sort(Collections.reverseOrder());
int sum = 0;
for (Integer i : list) {
answer++;
sum += i;
if (sum >= k) {
break;
}
}
return answer;
}
}
최초로 Map에 값을 넣는 과정은 다음과 같이 하면 더욱 편하게 값을 넣을 수 있다.
for (Integer i : tangerine) {
map.put(i, map.getOrDefault(i, 0) + 1);
}
d