347.前 K 个高频元素

【LetMeFly】347.前 K 个高频元素

力扣题目链接：https://leetcode.cn/problems/top-k-frequent-elements/

给你一个整数数组 nums 和一个整数 k ，请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。

 

示例 1:

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

示例 2:

输入: nums = [1], k = 1
输出: [1]

 

提示：

• 1 <= nums.length <= 105
• k 的取值范围是 [1, 数组中不相同的元素的个数]
• 题目数据保证答案唯一，换句话说，数组中前 k 个高频元素的集合是唯一的

 

进阶：你所设计算法的时间复杂度 必须 优于 O(n log n) ，其中 n 是数组大小。

方法一：哈希 + 排序

首先用哈希表统计每个数字出现的次数,次操作的总时间复杂度是O$(n)$（一共有$n$个元素）

之后将哈希表中的<数字,出现次数>键值对存入数组中，并对其排序，排序规则是“出现次数大的优先” $O(n\log n)$

最后，从前到后输出前$k$个排序后的“数字”即可

• 时间复杂度$O(n\log n)$，其中$n$是元素个数
• 空间复杂度$O(n)$

运行结果：时间/空间 超越了97.97%,19.06% 的用户

AC代码

C++

typedef pair<int, int> pii;

bool cmp(const pii& a, const pii& b) {
    return a.second > b.second;
}

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int, int> ma;
        for (int& t : nums) {
            ma[t]++;
        }
        vector<pii> v;
        for (auto thisPair : ma) {
            v.push_back(thisPair);
        }
        sort(v.begin(), v.end(), cmp);
        vector<int> ans;
        for (int i = 0; i < k; i++) {
            ans.push_back(v[i].first);
        }
        return ans;
    }
};

同步发文于CSDN，原创不易，转载请附上原文链接哦~
Tisfy：https://letmefly.blog.csdn.net/article/details/127131965