给定一个非空的整数数组,返回k个最频繁的元素
范例1:
输入: nums = [1,1,1,2,2,3],k = 2
输出:[1,2]
范例2:
输入: nums = [1],k = 1
输出:[1]
注意:
你可以假设ķ始终是有效的,1≤ ķ独特元素的数量≤。
您算法的时间复杂度必须好于O(n log n),其中n是数组的大小。
确保答案是唯一的,换句话说,前k个频繁元素的集合是唯一的。
您可以按任何顺序返回答案。
思路:
将所有的元素设置为pair对的,然后排序,最后输出前k个。
code:
class Solution {
public:
vector<int> topKFrequent(vector<int>& nums, int k) {
sort(nums.begin(), nums.end());
vector<pair<int,int> > times;
int count=0;
int tmp=nums[0];
for(auto t:nums)
{
if(tmp==t){
count++;
}
else{
times.push_back( make_pair(tmp,count) );
tmp=t;
count=1;
} //补上最后一个
}
times.push_back( make_pair(tmp,count) );
sort(times.begin(),times.end(), [](pair<int,int> p1, pair<int,int> p2){
return p1.second> p2.second ;
});
vector<int> res;
for(int i =0;i<k;++i)
res.push_back(times[i].first);
return res;
}
};