LeetCode 0347. 前 K 个高频元素

102 阅读1分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第1天,点击查看活动详情

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

力扣题目链接:leetcode.cn/problems/to…

给你一个整数数组 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)(一共有nn个元素)

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

最后,从前到后输出前kk个排序后的“数字”即可

  • 时间复杂度O(nlogn)O(n\log n),其中nn是元素个数
  • 空间复杂度O(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:letmefly.blog.csdn.net/article/det…