标签:nums int list 堆排序 countMap queue 347 new LeetCode
描述
给定一个非空的整数数组,返回其中出现频率前 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 是数组的大小。
解析
topN的问题都可以考虑使用堆。
这里的排序,稍微注意下。
代码
public List<Integer> topKFrequent(int[] nums, int k) { List<Integer> list = new ArrayList<>(k); Map<Integer, Integer> countMap = new HashMap<>(); for (int i = 0; i < nums.length && list.size() < k; i++) { countMap.put(nums[i], countMap.getOrDefault(nums[i], 0) + 1); } PriorityQueue<Integer> queue = new PriorityQueue<>((n1, n2) -> countMap.get(n1) - countMap.get(n2)); countMap.forEach((key, value) -> { queue.offer(key); if (queue.size() > k) { queue.poll(); } }); for (int i = 0; i < k; i++) { if (!queue.isEmpty()) { list.add(queue.poll()); } } return list; }
标签:nums,int,list,堆排序,countMap,queue,347,new,LeetCode 来源: https://www.cnblogs.com/fanguangdexiaoyuer/p/12088019.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。