前 K 个高频元素， 求Top N

 

 

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

示例 1:

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

输入: nums = [1], k = 1
输出: [1]
说明：

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

 

 本菜鸟自己的解题，对比官方思路。。。

//本菜鸟思路
public List<Integer> topKFrequent(int[] nums, int k) {
        Map<Integer,Integer> map = new HashMap();
        for(int i :nums){
            map.put(i, map.getOrDefault(i, 0) + 1);
        }
        Set<Integer> keys = map.keySet();
        Iterator<Integer> it = keys.iterator();
        int[][] array = new int[nums.length][2];
        for(int i =0; it.hasNext() && i < nums.length; i++){
            Integer key = it.next();
            array[i][0] = map.get(key);
            array[i][1] = key;
            
        }
        Arrays.sort(array,new Comparator<int[]>() {
            @Override
            public int compare(int[] o1, int[] o2) {
                return o2[0]- o1[0];
            }
        });
        List<Integer> ans =new ArrayList();
        for(int i=0; i < k; i++){
            ans.add(array[i][1]);
        }
        return ans;
    }

//官方思路    
    public List<Integer> topKFrequent(int[] nums, int k) {
        Map<Integer, Integer> map = new HashMap();
        for(int i :nums){
            map.put(i, map.getOrDefault(i,0) + 1);
        }
        List<Integer> ans = new ArrayList();
        Queue<Integer> prioQ = new PriorityQueue(new Comparator(){
            @Override
            public int compare(Object o1, Object o2){
                return map.get(o1) - map.get(o2);
            }
        });
        for(Integer i : map.keySet()){
            prioQ.offer(i);
            if(prioQ.size() > k){
                prioQ.poll();
            }
        }
        while(!prioQ.isEmpty()){
            ans.add(prioQ.poll());
        }
        Collections.reverse(ans);
        return ans;
    }