215.数组中的第K个最大元素/347. 前 K 个高频元素

2020-05-25

1.题目描述

数组中的第K个最大元素

2.题解

1.使用冒泡排序
2.使用快速排序
3.优先队列（内部是用堆来实现的）

3.代码

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        int l=nums.size();
        int i,j;
        for (i=0;i<k;i++){
            for (j=0;j<l-i-1;j++){
                if (nums[j]>nums[j+1]){
                    int t=nums[j];
                    nums[j]=nums[j+1];
                    nums[j+1]=t;
                }
            }
        }
        return nums[l-k];
    }
};

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        return fastsort(nums,0,nums.size()-1,k);
    }

    int fastsort(vector<int> vec,int l,int r,int k){
        int tmp=fun(vec,l,r);
        if (tmp==k-1) return vec[k-1];
        else if (tmp<k-1) return fastsort(vec,tmp+1,r,k);
        else return fastsort(vec,l,tmp-1,k);
    }

    int fun(vector<int> &vec,int l,int r){
        int tmp=vec[l];
        while (l<r){
            while (l<r&&vec[r]<=tmp) r--;
            vec[l]=vec[r];
            while (l<r&&vec[l]>=tmp) l++;
            vec[r]=vec[l];
        }
        vec[l]=tmp;
        return l;
    }
};

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        priority_queue <int,vector<int>,less<int> >que;
        int l=nums.size();
        for (int i=0;i<l;i++){
            que.push(nums[i]);
        }
        for (int i=0;i<k-1;i++){
            que.pop();
        }
        return que.top();
    }
};

347. 前 K 个高频元素

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        vector<int>res;
        map<int,int> mymap;
        int l=nums.size();
        priority_queue<pair<int,int>> myque;
        for (int i=0;i<l;i++){
            mymap[nums[i]]=0;
        }
        for (int i=0;i<l;i++){
            mymap[nums[i]]++;
        }
        for (map<int,int>::iterator it=mymap.begin();it!=mymap.end();it++){
            pair<int,int> tmp(it->second,it->first);
            myque.push(tmp);
        }
        for (int i=0;i<k;i++){
            res.push_back(myque.top().second);
            myque.pop();
        }
        return res;
    }
};