前 K 个高频元素--java

给你一个整数数组 nums 和一个整数 k ,请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。

示例 1:

输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]
示例 2:

输入: nums = [1], k = 1
输出: [1]

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/top-k-frequent-elements
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        int[] result = new int[k];//新建k长度的数组
        HashMap<Integer,Integer> map = new HashMap<>();//新建泛型的哈希表
        for(int i = 0;i<nums.length;i++){ //用一个哈希表获取每种元素出现的次数
            int num = nums[i];//num为key
            map.put(num,map.getOrDefault(num,0) + 1);//把每个键的默认值设为0
        }
        List<Map.Entry<Integer,Integer>> list=new ArrayList<Map.Entry<Integer,Integer>>(map.entrySet());//适合用set集合新建Entry对象的列表
        //从大到小排序,更新比较器
        Collections.sort(list,(o1,o2)->(o2.getValue()-o1.getValue()));
        //输出前k个元素
        for(int i=0;i<k;i++){
            result[i]=list.get(i).getKey();
        }
        return result;
    }
}
posted @ 2021-10-14 18:01  网抑云黑胶SVIP用户  阅读(36)  评论(0编辑  收藏  举报