347. Top K Frequent Elements (C++程序实现,两种方案)

347. Top K Frequent Elements (C++程序实现,两种方案)_第1张图片

解题方案一:

先对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;
    }
};


你可能感兴趣的:(LeetCode,C++)