LeetCode Top K 问题

最新推荐文章于 2024-07-18 23:39:51 发布

Z_Y_D_

最新推荐文章于 2024-07-18 23:39:51 发布

阅读量2.3k

点赞数

分类专栏： leetcode 文章标签： LeetCode TopK

本文链接：https://blog.csdn.net/Z_Y_D_/article/details/84646443

版权

本文探讨如何解决LeetCode中找到数组出现频率前k高的元素的问题。提出了两种方法：基于快排和基于堆排序。强调在处理大量数据时，堆排序（维护最小堆）是更优的选择，其时间复杂度为O(n*logK)。

摘要由CSDN通过智能技术生成

给定一个非空的整数数组，返回其中出现频率前 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 是数组的大小。

方法一：基于快排当划分到下标为（length-k）后，其后元素即为前K大的元素

class Solution {
    public List<Integer> topKFrequent(int[] nums, int k) {
        List<Integer> res=new ArrayList<>();
//      先用hashmap统计词频
        HashMap<Integer,Integer> map=new HashMap<>();
        for(int i=0;i<nums.length;i++){
            if(map.containsKey(nums[i])) {
                map.put(nums[i],map.get(