leetcode 347. Top K Frequent Elements

最新推荐文章于 2025-12-03 23:35:37 发布

转载最新推荐文章于 2025-12-03 23:35:37 发布 · 331 阅读

文章标签：

#leetcode

Leetcode 专栏收录该内容

48 篇文章

订阅专栏

本文介绍了一种使用HashMap和优先级队列实现的算法，该算法可在给定一个非空整数数组的情况下返回出现频率最高的k个元素。该算法的时间复杂度优于O(n log n)，并提供了完整的Java代码实现。

原文链接http://blog.youkuaiyun.com/li563868273/article/details/51484932

题目

Given a non-empty array of integers, return the k most frequent elements.

For example,
Given [1,1,1,2,2,3] and k = 2, return [1,2].

Note:
You may assume k is always valid, 1 ≤ k ≤ number of unique elements.
Your algorithm’s time complexity must be better than O(n log n), where n is the array’s size.

解

public class Solution {
    public List<Integer> topKFrequent(int[] nums, int k) {
        //利用hashMap，和优先级队列。
        List<Integer> res=new ArrayList<Integer>();

        Map<Integer,Integer> map=new HashMap<Integer,Integer>();
        if(k>nums.length||k<1){
            return res;
        }
        for(int num:nums){
            if(map.containsKey(num)){
                map.put(num,map.get(num)+1);
            }else{
                map.put(num,1);
            }
        }
        ArrayList<Map.Entry<Integer,Integer>> list = new ArrayList<Map.Entry<Integer, Integer>>(map.entrySet());
        Collections.sort(list, new Comparator<Map.Entry<Integer, Integer>>() {
            @Override
            public int compare(Map.Entry<Integer, Integer> o1, Map.Entry<Integer, Integer> o2) {
                return o2.getValue() - o1.getValue();
            }
        });
        int count = 0;
        while (count < k)
            res.add(list.get(count++).getKey());
        return res;
    }

}