您的位置:首页 > 产品设计 > UI/UE

[LeetCode] Top K Frequent Elements

2016-09-21 21:18 295 查看
Given a non-empty array of integers, return the k most
frequent elements.
For example,
Given [1,1,1,2,2,3] and
k = 2, return [1,2].

Note: 

You may assume k is always valid, 1 ≤ k ≤ number of unique elements.
Your algorithm's time complexity must be better than O(n log n), where n is the array's size.
思路:
使用map 存储计算每个元素的出现次数,将map中的<键,值>放入vector中,使用sort函数降序排列,输出前k个对应的数字
bool  cmp (const pair<int, size_t> &x, const pair<int,  size_t> &y)
{
return x.second > y.second;
}
class Solution {
public:
vector<int> topKFrequent(vector<int>& nums, int k)
{
vector<int> res;
unordered_map<int,  size_t> m;
for(int i = 0; i < nums.size(); i++)
{
++m[nums[i]];
}
vector<pair<int,  size_t>> vec(m.begin(),m.end());
sort(vec.begin(), vec.end(),  cmp);
for(int i = 0; i < k; i++)
{
res.push_back(vec[i].first);
}
return res;
}

};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: