leetcode练习前 K 个高频元素(优先队列和哈希表的使用)

给定一个非空的整数数组，返回其中出现频率前 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 是数组的大小。
思路：
用哈希表第一个参数是数组元素第二个出现的次数。之后遍历keyset用优先队列放进去。我们取的是前面k大的元素但在优先队列里面应该是按照频率从小到大排序(排序的指标需要我们自己创建一个比较器,因为我们不是基于自然顺序而是在哈希表中value的值)。之后用数组列表一次将队列弹出。

class Solution {
    public List<Integer> topKFrequent(int[] nums, int k) {
        List<Integer> list=new ArrayList<>();
        Map<Integer,Integer> map=new HashMap<>();
        for(int i:nums) {
        	map.put(i,map.getOrDefault(i, 0)+1);
        }
        Comparator c=new Comparator<Integer>() {
        	public int compare(Integer i,Integer j) {
        		return map.get(i).compareTo(map.get(j));
        	}
        };
        PriorityQueue<Integer> queue=new PriorityQueue<>(k,c);	
        Set<Integer> set=map.keySet();
        for(int key:set) {
        	if(queue.size()<k)		//集合没有满，直接放进去
        		queue.offer(key);
        	else if(map.get(queue.peek())<map.get(key)) {	//比队列中的最小元素大那么对头出队，放一个到队列中
        		queue.poll();
        		queue.offer(key);
        	}
        }
        for(int i=0;i<k;i++) {
        	list.add(queue.poll());		//弹出到数组列表
        }
      return list;
    }
    public static void main(String[] ARGS) {
    	int nums[]= {4,1,-1,2,-1,2,3};
    	List<Integer> list=new Solution().topKFrequent(nums, 4);
    	for(int i:list) {
    		System.out.print(i+"   ");
    	}
    }
}