标签：vector num int res 元素 times second 347 高频

要求：返回前k个高频元素，时间不超O(nlogn)，注意是指k个不同的数而不是k个频率
思路：
法一：首先记录次数数组，由于时间限制不能排序，故对该数组用小顶堆，初始化堆O(klogk)，调整O(n-k)(logk)，空间O(n)

class Solution {
public:
    struct cmp{
	    bool operator()(pair<int,int> &a,pair<int,int> &b){
		    return a.second>b.second;
	    }
    };
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int,int> times;
        int n=nums.size();
        for(int i=0;i<n;i++)
            times[nums[i]]++;
        priority_queue<pair<int,int>,vector<pair<int,int>>,cmp> minheap;
        int num=0;
        for(auto a:times){
            if(num<k){
                minheap.push(a);
                num++;
            }
            else if(a.second>minheap.top().second){
                minheap.pop();
                minheap.push(a);
            }
        }
        vector<int> res;
        while(!minheap.empty()){
            res.push_back(minheap.top().first);
            minheap.pop();
        }
        return res;
    }
};

法二：快排的partition函数找次数数组里第n-k个，随便写写，时间在O(n)~O(n^2)，空间O(n)

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int,int> hash;
        int n=nums.size();
        for(int i=0;i<n;i++)
            hash[nums[i]]++;
        vector<int> times;
        for(pair<int,int> a:hash)
            times.push_back(a.second);
        n=times.size();
        k=n-k;
        int l=0,r=n-1;
        int num=0;
        while(true){
            num=partition(times,l,r);
            if(num==k)break;
            else if(num<k)l=num+1;
            else r=num-1;
        }
        cout<<num;
        vector<int> res;
        for(pair<int,int> a:hash)
            if(a.second>=times[num])
                res.push_back(a.first);
        return res;
    }
    int partition(vector<int> &times,int l,int r){
        int i=l,j=r;
        int pivot=times[l];
        while(i<j){
            while(i<j&&times[j]>=pivot)
                j--;
            times[i]=times[j];
            while(i<j&&times[i]<=pivot)
                i++;
            times[j]=times[i];
        }
        times[i]=pivot;
        return i;
    }
};

法三：桶排序，空间换时间，每个频率一个vector，注意一下vector下表，时间O(n)，空间O(n)

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int,int> times;
        for(auto a:nums)
            times[a]++;
        int maxfreq=0;
        for(auto a:times)
            maxfreq=max(maxfreq,a.second);
        vector<vector<int>> buckets(maxfreq);
        for(auto a:times)
            buckets[a.second-1].push_back(a.first);
        vector<int> res;
        int num=0;
        for(int i=maxfreq-1;num<k&&i>=0;i--){
            if(buckets[i].empty())continue;
            for(auto a:buckets[i]){
                res.push_back(a);
                num++;
            }
        }
        return res;
    }
};

标签：vector,num,int,res,元素,times,second,347,高频
来源： https://blog.csdn.net/cx_cs/article/details/120524405

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。