题目
- 不同整数的最少数目
给你一个整数数组 arr 和一个整数 k 。现需要从数组中恰好移除 k 个元素,请找出移除后数组中不同整数的最少数目。
示例 1:
输入:arr = [5,5,4], k = 1
输出:1
解释:移除 1 个 4 ,数组中只剩下 5 一种整数。
示例 2:
输入:arr = [4,3,1,1,3,3,2], k = 3
输出:2
解释:先移除 4、2 ,然后再移除两个 1 中的任意 1 个或者三个 3 中的任意 1 个,最后剩下 1 和 3 两种整数。
提示:
1 <= arr.length <= 10^5
1 <= arr[i] <= 10^9
0 <= k <= arr.length
提交记录
这种方法效率比较低。
代码
import java.util.Map;
class Solution {
public int findLeastNumOfUniqueInts(int[] arr, int k) {
// 数组转Map,key(数组元素),value(元素出现的次数)
Map<Integer,Integer> map = new HashMap<Integer, Integer>();
for(int i=0; i<arr.length; i++){
if(map.containsKey(arr[i])){
int val = map.get(arr[i]);
map.put(arr[i],val+1);
}else{
map.put(arr[i],1);
}
}
// 按照value值(数组中元素重复出现的次数)从小到大排序
HashMap<Integer, Integer> sortMap = new LinkedHashMap<Integer, Integer>();
List<Map.Entry<Integer, Integer>> mapList = map.entrySet()
.stream()
.sorted((p1,p2)->p1.getValue().compareTo(p2.getValue()))
.collect(Collectors.toList());
//遍历集合,将排好序的键值对Entry<K,V>放入新的map并返回。
mapList.forEach(ele->sortMap.put(ele.getKey(), ele.getValue()));
// map转list
List<Integer> list = new ArrayList<>();
for(Map.Entry<Integer,Integer> entry:sortMap.entrySet()){
for(int i=0;i<entry.getValue();i++){
list.add(entry.getKey());
}
};
// 删除排序后的list前k个元素
for (int y = 0; y < k; y++) {
list.remove(0);
}
// 去重
TreeSet<Integer> set = new TreeSet<>(list);
return set.size();
}
}