数组中第k大的数字

​

给定整数数组 nums 和整数 k，请返回数组中第 k 个最大的元素。

请注意，你需要找的是数组排序后的第 k 个最大的元素，而不是第 k 个不同的元素。

示例 1:

1
2

输入: [3,2,1,5,6,4] 和 k = 2
输出: 5

示例 2:

1
2
3
4

输入: [3,2,3,1,2,4,5,5,6] 和 k = 4
输出: 4

提示：

1
2

1 <= k <= nums.length <= 104
-104 <= nums[i] <= 104

注意：本题与主站 215 题相同： https://leetcode-cn.com/problems/kth-largest-element-in-an-array/ 

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        int n = nums.size();
        //第k大 => n-k 
        return quick_find(nums,n-k+1 ,0, n-1);
    }
    int quick_find(vector<int> &nums,int pos,int l,int r) {
        if( l>= r) return nums[l];
        int x = nums[(l+r)/2];
        int i=l-1,j = r+1;
        while(i<j) {
            do i++; while(i<=r && nums[i] <x);
            do j--; while(j>=l && nums[j] > x);
            if (i<j) {
                swap(nums[i],nums[j]);
            }
        } 
        
        // if(j-l==pos) return nums[j];
        if(j-l+1>= pos) {
            return quick_find(nums,pos,l,j);
        }
        return quick_find(nums,pos-(j-l+1),j+1,r);
    }  
        
};

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        int n = nums.size();
        //第k大 => n-k 
        return quick_find(nums,n-k+1 ,0, n-1);
    }
    int quick_find(vector<int> &nums,int pos,int l,int r) {
        if( l>= r) return nums[l];
        int x = nums[(l+r)/2];
        int i = l-1, j = r+1;
        while(i<j) {
            i++;while(i<=r && nums[i]<x) i++  ;
            j--;while(j>=l && nums[j] > x) j-- ;
            if (i<j) swap(nums[i],nums[j]);
        }
        
        // if(j-l==pos) return nums[j];
        if(j-l+1>= pos) {
            return quick_find(nums,pos,l,j);
        }
        return quick_find(nums,pos-(j-l+1),j+1,r);
    }  
        
};