[Java] 138475 귤 고르기
📝 귤 고르기
1. 문제 요약
- 귤 k개를 고를 때 크기가 서로 다른 종류의 수의 최솟값을 return
2. 정답코드
import java.util.*;
class Solution {
public int solution(int k, int[] tangerine) {
HashMap<Integer, Integer> countMap = new HashMap<>();
for (int i = 0; i < tangerine.length; i++) {
countMap.put(tangerine[i], countMap.getOrDefault(tangerine[i], 0) + 1);
}
List<Integer> counts = new ArrayList<>(countMap.values());
counts.sort(Collections.reverseOrder());
int picked = 0;
int kinds = 0;
for (int c : counts) {
picked += c;
kinds++;
if (picked >= k) break;
}
return kinds;
}
}
3. 설명
- 해시맵에 귤의 종류, 개수를 매핑한다.
- 귤의 개수를 담을 List count를 선언하고 내림차순 정렬한다.
- 고른 귤 개수, 종류를 담을 변수 picked, kinds 를 선언한다.
- count를 하나씩 순회하며 picked에 갯수를 누적하고, 종류+1을 한다.
- 만약 고른 귤 개수가 k보다 크거나 같으면 멈춘다.
4. 오답 / 어려웠던 점
- ArrayList만들고 map의 밸류와 매핑하는 법 new ArrayList<>(countMap.values());
- 오름차순 정렬 Collections.sort(counts)
- 내림차순 정렬 counts.sort(Collections.reverseOrder())