解题方案一:
先对nums中的元素排序,然后将元素及元素出现的次数存入二元组vector<pair<int,int>> a;中,再次对a排序(pair<int,int>第一个存nums[i]出现的次数,第二个存nums[i]),a中末尾的k个数对中的pair.second即为所求。
class Solution {
public:
vector<int> topKFrequent(vector<int>& nums, int k) {
vector<pair<int,int>> a;
vector<pair<int,int>>::iterator it;
sort(nums.begin(),nums.end());
for(int i=0;i<nums.size();++i){
int count=1;
while(i+1<nums.size()&&i<nums.size()&&nums[i+1]==nums[i]){
++count;
++i;
}
a.push_back(pair<int,int>(count,nums[i]));
}
sort(a.begin(),a.end());
int i=a.size()-1,j=0;
nums.clear();
while(j<k&&i>=0){
nums.push_back(a[i].second);
++j;
--i;
}
return nums;
}
};
解题思路二:
因为map和multimap容器存储的元素对,会自动根据key值自动排序,所以只要统计出每个元素出现的次数,然后将次数作为key值,元素作为value值存入multimap中。直接遍历multimap最后k个数对就得到k个出现频率最高的元素。
map<int,int>存储数组中的元素nums[i]及其对应出现的次数。然后将map中的数对(key,value)对调存储于multimap<int,int>,用迭代器从后往前遍历multimap。
class Solution {
public:
vector<int> topKFrequent(vector<int>& nums, int k) {
vector<int> v;
map<int,int> one;
multimap<int,int> two;
map<int,int>::iterator i1;
multimap<int,int>::iterator i2;
for(int i=0;i<nums.size();++i){
int key=nums[i];
i1=one.find(key);
if(i1!=one.end()){//找到
int k=(*i1).second;
++k;
one.erase(i1);
one[key]=k;
}
else//没找到
one[key]=1;
}
for(i1=one.begin(),i2=two.begin();i1!=one.end();++i1){
two.insert(pair<int,int>((*i1).second,(*i1).first));
}
int j=0;
i2=two.end();
--i2;
while(j<k){
if(i2!=two.begin()){
v.push_back((*i2).second);
++j;
--i2;}
else{
v.push_back((*i2).second);
break;
}
}
return v;
}
};