leetcode 347 前K个高频元素

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

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

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

思路:首先应该考虑用HashMap来做,建立数字和其出现次数的映射,然后再按照出现次数进行排序。我们可以用堆排序来做,使用一个最大堆来按照映射次数从大到小排列,在java中使用PriorityQueue来实现。

public List topKFrequent(int[] nums, int k) {
    //第一步:存放值 和值出现的次数
    Map map=new HashMap<>();
    for (int num : nums) {
        if(map.get(num) == null){
            map.put(num, 1);
        }else{
            map.put(num, map.get(num) + 1);
        }
    }
    //第二步:生成优先队列
    PriorityQueue queue = new PriorityQueue<>(new Comparator() {
        @Override
        public int compare(Integer a, Integer b) {
            return map.get(b) - map.get(a);
	    }
	});
    for(Integer key : map.keySet()){
        queue.add(key);
    }
    List list = new ArrayList<>();
    for(int i = 0; i < k; i++){
        list.add(queue.remove());
    }
    return list;
}

 

你可能感兴趣的:(leetecode)