数组第k最大

数组第k最大

题目

215. 数组中的第K个最大元素

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

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

你必须设计并实现时间复杂度为 O(n) 的算法解决此问题。

示例 1:

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

示例 2:

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

提示：

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

思路

这里我们可以用快速排序来做，我们做快速排序的时候，会根据一个基准值然后进行分区，比基准大的都放到右边，把基准小的都放到左边，最后我们会把基准放到相应的位置，此时会有一个索引值，这时我们可以比较k值，如果k小于等于索引值，那么我们就继续进行左边的排序，否则进行右边的排序，这样我们就可以把时间减半。直接来看代码吧

题解

class Solution {
    public int findKthLargest(int[] nums, int k) {
        int n = nums.length;
        return selectOne(nums,0,n-1,n-k); // 因为题目是要找第k最大的所以我们正序排序的话就要进行一个n-k的一次次序倒转。才能满足第k最大
    }
    public int selectOne(int[] nums,int l ,int r,int k){
        if(l == r ){
            return nums[k];
        }
        int x = nums[l],i=l-1,j=r+1;
        while(i<j){
            do i++;while(nums[i] <x);
            do j--; while(nums[j] > x);
            if(i<j){
                int tmp = nums[i];
                nums[i] = nums[j];
                nums[j] = tmp;
            }
        }
        int tmp = nums[l];
        nums[l] = nums[j];
        nums[j] = tmp;
        if(j >= k) return selectOne(nums,l,j,k);
        else return selectOne(nums,j+1,r,k);
    }
}

点击链接：我正在「编程导航」和朋友们讨论有趣的话题，你⼀起来吧？

也可以加我QQ（2837468248）咨询说明来意！