码迷,mamicode.com
首页 > 其他好文 > 详细

leetcode347 前k个高频元素(Medium)

时间:2020-07-18 19:48:04      阅读:59      评论:0      收藏:0      [点我收藏+]

标签:ati   turn   i++   ++   频率   push   数字   最小堆   rbegin   

题目来源:leetcode347 前k个高频元素

题目描述:

给定一个非空的整数数组,返回其中出现频率前?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?是数组的大小。
题目数据保证答案唯一,换句话说,数组中前 k 个高频元素的集合是唯一的。
你可以按任意顺序返回答案。

解题思路:

用map保存数字出现的次数,并进行排序。

class Solution {
public:
    static bool cmp(pair <int,int> p1,pair<int,int> p2){
        return p1.second>p2.second;
    }
    vector<int> topKFrequent(vector<int>& nums, int k) {
        map<int,int> m;
        int i;
        for(i=0;i<nums.size();i++){
            m[nums[i]]++;
        }
        vector<int> ans;
        vector<pair<int,int>> res(m.begin(),m.end());
        sort(res.begin(),res.end(),cmp);
        for(i=0;i<k;i++){
            ans.push_back(res[i].first);
        }
        return ans;
    }
};

或者也可以用最小堆。

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        map<int,int> m;
        priority_queue<int,vector<pair<int,int>>,greater<pair<int,int>>> pq;
        for(auto i:nums){
            m[i]++;
        }
        for(auto it:m){
            if(pq.size()<k){
                pq.push(make_pair(it.second,it.first));
            }
            else{
                if(it.second>pq.top().first){
                    pq.pop();
                    pq.push(make_pair(it.second,it.first));
                }
            }
        }
        vector<int> ans;
        while(!pq.empty()){
            ans.push_back(pq.top().second);
            pq.pop();
        }
        return vector<int>(ans.rbegin(),ans.rend());
    }
};

leetcode347 前k个高频元素(Medium)

标签:ati   turn   i++   ++   频率   push   数字   最小堆   rbegin   

原文地址:https://www.cnblogs.com/yjcoding/p/13336438.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!