高级数据结构

最新推荐文章于 2022-07-23 12:05:34 发布

梦世

最新推荐文章于 2022-07-23 12:05:34 发布

阅读量343

点赞数

分类专栏：数据结构与刷题

本文链接：https://blog.youkuaiyun.com/qq_34949782/article/details/107227521

版权

数据结构与刷题专栏收录该内容

18 篇文章

订阅专栏

优先队列

LetCode：347. 前 K 个高频元素

优先队列

向上：新增节点

向下：堆顶被取

初始化大小为n的堆，时间复杂度：O（n）

LetCode：347. 前 K 个高频元素

class Solution {
 	public static int[] topKFrequent(int[] nums, int k) {
		int[] result = new int[k];
		Map<Integer, Integer> map = new HashMap<Integer, Integer>();
		for (int temp : nums) {
			if (map.containsKey(temp)) {
				map.put(temp, map.get(temp) + 1);
			} else {
				map.put(temp, 1);
			}
		}

		Queue<Map.Entry<Integer, Integer>> priQueue = new PriorityQueue<Map.Entry<Integer, Integer>>(k,
				(o1,o2) -> o2.getValue() - o1.getValue() );

		Set<Map.Entry<Integer, Integer>> set = map.entrySet();
		Iterator<Map.Entry<Integer, Integer>> iter = set.iterator();
		while (iter.hasNext()) {
			Map.Entry<Integer, Integer> me = iter.next();
			priQueue.add(me);
		}

		for (int i = 0; i < k; i++) {
			result[i] = priQueue.poll().getKey();
		}

		return result;
	}
}