给你一个整数数组 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 两种整数。
来源于leetcode上题目
解题思路:
解决该问题的关键是找出数组中出现次数较少的整数,因此要构造一个根据整数出现次数排序的列表即ArrayList<Map.Entry<Integer,Integer>>或ArrayList<Integer[]> ,其中Map.Entry<Integer,Integer>的第一个Integer存放的是整数,第二个Integer存放的是整数出现的次数,而Integer[]中有两个数,与上同样,第一个存放的是整数,第二个存放的是整数出现的次数。通过Collection.sort()来排序,其中自定义排序方法。
代码实现:
class Solution {
public int findLeastNumOfUniqueInts(int[] arr, int k) {
HashMap<Integer,Integer> new_arr = new HashMap();
// 完成数据的统计
for (int i = 0; i < arr.length; i++) {
int count = new_arr.getOrDefault(arr[i], 0)+1;
new_arr.put(arr[i],count);
}
// 排序
ArrayList<Map.Entry<Integer,Integer>> sortArr = new ArrayList(new_arr.entrySet());
Collections.sort(sortArr, new Comparator<Map.Entry<Integer, Integer>>() {
@Override
public int compare(Map.Entry<Integer, Integer> o1, Map.Entry<Integer, Integer> o2) {
return o1.getValue() - o2.getValue();
}
}
);
// 进行移除
int surplus = sortArr.size();
for (int i = 0; i < sortArr.size() && k >0; i++) {
if(sortArr.get(i).getValue() <= k){
surplus--;
k -= sortArr.get(i).getValue();
}
}
return surplus;
}
}