leetcode题目1481. 不同整数的最少数目

题目

1481. 不同整数的最少数目
      给你一个整数数组 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();
    }
}