Leetcode刷题笔记 347. 前 K 个高频元素

时间：2020年9月7日
知识点：hashmap、最小堆
题目链接：https://leetcode-cn.com/problems/top-k-frequent-elements/

题目
给定一个非空的整数数组，返回其中出现频率前 k 高的元素。

示例1
输入：
nums = [1,1,1,2,2,3], k = 2
输出：
[1,2]

示例2
输入：
nums = [1], k = 1
输出：
[1]

说明：

1. 你可以假设给定的 k 总是合理的，且 1 ≤ k ≤ 数组中不相同的元素的个数。
2. 你的算法的时间复杂度必须优于 O(n log n) , n 是数组的大小。
3. 题目数据保证答案唯一，换句话说，数组中前 k 个高频元素的集合是唯一的。
4. 你可以按任意顺序返回答案。

解法

1. 首先想到用unordered_map 建立数字和个数的映射
2. 对频率排序
3. 对频率排序有两种方法,
   1. 把map转成vector,包含数字和个数,全部排列,取前k个,时间复杂O(N²)
   2. 把map转成大小为堆,堆的大小为k,时间复杂度为O(Nlogk)
4. 取出频率对应的数字

代码

#include <stdio.h>
#include <iostream>
#include <vector>
#include <unordered_map>
#include <queue>
using namespace std;
/*
//hashmap+快排
bool cmp(pair<int,int> x, pair<int,int> y){
    return x.second>y.second;
};
class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        vector<int> ans;
        unordered_map<int,int> m;
        int n = nums.size();
        vector<pair<int,int>> v(n);
        for(int i=0;i<nums.size();i++){
            m[nums[i]] ++;
        }
        for(unordered_map<int, int>::iterator i=m.begin();i!=m.end();i++){
            v.push_back(*i);
        }
        sort(v.begin(),v.end(),cmp);
        for(int i=0;i<k;i++)
            ans.push_back(v[i].first);
        return ans;
    }
};
*/
//hashmap+堆
class Solution {
public:
    struct cmp{
        bool operator()(pair<int, int> x,pair<int, int> y){
            return x.second > y.second;
        }
    };
    vector<int> topKFrequent(vector<int>& nums, int k) {
        vector<int> ans(k,0);
        unordered_map<int,int> m;
        priority_queue<pair<int, int>,vector<pair<int, int>>,cmp> q;
        int n = nums.size();
        for(int i=0;i<nums.size();i++){
            m[nums[i]] ++;
        }
        for(auto i=m.begin();i!=m.end();i++){
            q.push(*i);
            if(q.size() > k)
                q.pop();
        }
        for(int i=k-1;i>=0;i--){
            ans[i] = q.top().first;
            q.pop();
        }
        return ans;
    }
};
int main()
{
    vector<int> nums{1,1,1,2,2,3};
    Solution s;
    int k = 2;
    vector<int> ans = s.topKFrequent(nums, k);
    for(int i=0;i<k;i++)
        cout<<ans[i]<<endl;
    return 0;
}

今天也是爱zz的一天哦！