数组中的第K个最大元素（3种解法）

最新推荐文章于 2024-04-08 08:28:48 发布

_vegetable_

最新推荐文章于 2024-04-08 08:28:48 发布

阅读量85

点赞数 1

分类专栏： code 文章标签：算法排序算法 leetcode

本文链接：https://blog.csdn.net/bkhyv/article/details/130089812

版权

code 专栏收录该内容

12 篇文章 0 订阅

订阅专栏

基于快速排序的快速选择

class Solution {
public:
    int quickSelect(vector<int>& a, int l, int r, int index) { //快速排序，可以看官方的解答思路，但是不要看他的代码，太乱了
        int q = partition(a, l, r);
        if (q == index)
            return a[q];
        else 
        {
            if(q < index)
                return quickSelect(a, q + 1, r, index);
            else
                return quickSelect(a, l, q - 1, index);
        }
    }

    inline int partition(vector<int>& a, int left, int right) {
        int x = a[left];
        while(left<right)
        {
            while(left<right && x<=a[right])
                --right;
            swap_(a, left, right);
            while(left<right && x>=a[left])
                ++left;
            swap_(a, left, right);
        }
        return left;
    }

    inline void swap_(vector<int>& a, int m, int n) {
        int temp = a[n];
        a[n] = a[m];
        a[m] = temp;
    }

    int findKthLargest(vector<int>& nums, int k) {
        return quickSelect(nums, 0, nums.size() - 1, nums.size() - k);
    }
};

借助Map，但是效率不高

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        map<int,int,greater<int>> u_map;
        for(const auto& n: nums)
            u_map[n]++;
        //vector<pair<int,int>> v_val_times{u_map.begin(), u_map.end()}
        for(map<int,int>::const_iterator it = u_map.begin(); it != u_map.end(); ++it)
        {
            int size = (*it).second;
            k -= size;
            if(k <= 0)
                return (*it).first;
        }
        return 0;
    }
};

借助优先队列，大顶堆进行求解

class Solution {
  public:
    int findKthLargest(vector<int>& nums, int k) {
        priority_queue<int> q(nums.begin(), nums.end());

        for(int i = 0; i < k - 1; i++) {
            q.pop();
        }
        return q.top();
    }
};