LeetCode 中级 - Top K Frequent Elements

Top K Frequent Elements

给定一个非空的整数数组，返回其中出现频率前 k 高的元素。

例如，

给定数组 [1,1,1,2,2,3] , 和 k = 2，返回 [1,2]。

注意：

• 你可以假设给定的 k 总是合理的，1 ≤ k ≤ 数组中不相同的元素的个数。
• 你的算法的时间复杂度必须优于 O(n log n) , n 是数组的大小。

分析

关于O(nlogn)的时间复杂度，一致都很懵，只知道优于O(n^2)差于O(n)，于是乎这道题直接就放弃考虑时间复杂度了

思路是用HashMap记录<元素1，频次1>，然后将其转化为数组，数组下标记录频次，值为元素。这样我们倒序遍历k个元素，即可得到前K高的元素

代码

    class Solution {
        //HashMap记录<元素，频次>键值对
        //用数组下标映射元素的出现次数
        //两次遍历数组即可
        //缺点在于需要额外的数组空间，比较耗空间
        public List<Integer> topKFrequent(int[] nums, int k) {
            List<Integer>[] fre = new ArrayList[nums.length+1];
            List<Integer> list = new ArrayList<>();
            HashMap<Integer,Integer> map = new HashMap<>();
            for(int i:nums){
                if(map.containsKey(i)){
                    map.put(i,map.get(i)+1);
                }else{
                    map.put(i,1);
                }
            }
            Iterator iter = map.entrySet().iterator();
            while(iter.hasNext()){
                Map.Entry<Integer,Integer> entry = (Map.Entry)iter.next();
                if(fre[entry.getValue()]==null){
                    ArrayList<Integer> temp = new ArrayList<>();
                    temp.add(entry.getKey());
                    fre[entry.getValue()] = temp;
                }else{
                    fre[entry.getValue()].add(entry.getKey());
                }

            }
            for(int i=nums.length;i>=0&&k>list.size();i--){
                if(fre[i]!=null){
                    list.addAll(fre[i]);
                }
            }
            return list;
        }
    }