Leetcode_前k个高频元素

思路大概是将数组中所有元素的频率记下来,然后统一使用最小堆来管理,将根节点与下一个元素比较,将频率更的高元素插入最小堆,同时进行堆排序操作,以此进行管理

在这里发现原来c++有优先队列priority_queue可以达到相同的效果,优先队列将权值最大的放在队列的出口处,这样就完成了类似的效果

原算法:

class Solution {

public:

    int sift_down(vector<pair<int,int>> &v,int k){

        int father=0;

        int child=2*father+1;

        pair<int,int> temp=v[0];

        while(child<k){

            if((child+1<k)&&(v[child].first>v[child+1].first))

                child++;

            if(temp.first>v[child].first){

                v[father]=v[child];

                father=child;

                child=father*2+1;

            }

            else

                break;

        }

        v[father]=temp;

        return 0;

    }

    

    vector<int> topKFrequent(vector<int>& nums, int k) {

        vector<pair<int,int>> v;

        for(int i=0;i<k;i++){

            v.push_back(make_pair(0,0));

        }

        map<int,int> m;

        for(vector<int>::iterator it=nums.begin();it!=nums.end();it++){

            if(m.find(*it)->second==0)

                m.insert(make_pair(*it,1));

            else

                m.find(*it)->second++;            

        }

        for(map<int,int>::iterator it=m.begin();it!=m.end();it++){

            if(it->second>v[0].first){

                v[0]=make_pair(it->second,it->first);

                sift_down(v,k);   

            }

        }

        vector<int> v1;

        for(vector<pair<int,int>>::iterator i=v.begin();i!=v.end();i++)

            v1.push_back(i->second);

        return v1;

    }

    

};

改进算法,写的很漂亮:

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
         unordered_map<int,int> m;
        priority_queue<pair<int,int>> p;
        vector<int> res;
        for(int num:nums)
        {
            ++m[num];
        }
        for(auto a:m)
        {
            p.push({a.second,a.first});
        }
        for(int i=0;i<k;++i)
        {
            res.push_back(p.top().second);
            p.pop();
        }
        return res;
    }
};

点赞