215. Kth Largest Element in an Array （Medium）

最新推荐文章于 2020-12-02 19:35:32 发布

YYYue502

最新推荐文章于 2020-12-02 19:35:32 发布

阅读量258

点赞数

分类专栏： LeetCode 文章标签：快速排序

本文链接：https://blog.csdn.net/wyh476901857/article/details/60789706

版权

LeetCode 专栏收录该内容

28 篇文章 0 订阅

订阅专栏

原题目：
　　Find the kth largest element in an unsorted array. Note that it is the kth largest element in the sorted order, not the kth distinct element.

　　For example,
　　Given [3,2,1,5,6,4] and k = 2, return 5.

题目大意如下：

　　给你一个未排序的数组找到数组当中第K大的那个数，注意是返回有序数组当中第K大的那个数，并不是第K远的那个数。

解题思路：

　　这道题题目意思当然是很简单的，做法也有很多种，但我们需要做的就是要找到时间复杂度最小的那一种方法（最后可以知道最小复杂度为稳定的O(n)），在这里我顺便也把其他方法简单列举一下。
　　1.快速排序（利用sort函数，复杂度为O(nlogn)。）

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        sort(nums.begin() , nums.end()) ;
        return nums[nums.size() - k] ;
    }
};

　　2.利用快速排序的选择思想（最坏情况O(n²)，最好情况O(n)）
　　

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        int left = 0, right = nums.size() - 1;
        while (1) {
            int pos = partition(nums, left, right);
            //如果选择的pivot符合条件就返回
            if (pos == k - 1) return nums[pos];
            //选择的pivot太小了，我们取大的那部分继续
            if (pos > k - 1) right = pos - 1;
            //选择的pivot太大了，取小的那部分继续
            else left = pos + 1;
        }
    }

    //这个化分区间的方法可以数组划分成三个区间，左边的不小于大pivot，右边的不大于pivot
    int partition(vector<int>& v , int left ,int right){
        int pivot = v[left];
        int l = left + 1, r = right;
        while (l <= r) {
            if (v[l] < pivot && v[r] > pivot)
                swap(v[l++], v[r--]);
            if(v[l] >= pivot) l++; 
            if(v[r] <= pivot) r--;
        }
        swap(v[left], v[r]);
        return r;
    }
};

　　3.利用随机选取pivot优化（保证时间复杂度为O(n)）
　　



    int partition(vector<int>& v , int left ,int right){
        //进行随机选取pivot
        srand((unsigned)time(NULL)) ;
        int random = rand()%(right - left + 1) + left ;
        swap(v[left] , v[random]) ;

        int pivot = v[left];
        int l = left + 1, r = right;
        while (l <= r) {
            if (v[l] < pivot && v[r] > pivot)
                swap(v[l++], v[r--]);
            if(v[l] >= pivot) l++; 
            if(v[r] <= pivot) r--;
        }
        swap(v[left], v[r]);
        return r;
    }
};