#yyds干货盘点# LeetCode 腾讯精选练习 50 题：数组中的第K个最大元素

Posted 2022-11-17

题目：

给定整数数组 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

代码实现：

class Solution 
    Random random = new Random();

    public int findKthLargest(int[] nums, int k) 
        return quickSelect(nums, 0, nums.length - 1, nums.length - k);
    

    public int quickSelect(int[] a, int l, int r, int index) 
        int q = randomPartition(a, l, r);
        if (q == index) 
            return a[q];
         else 
            return q < index ? quickSelect(a, q + 1, r, index) : quickSelect(a, l, q - 1, index);
        
    

    public int randomPartition(int[] a, int l, int r) 
        int i = random.nextInt(r - l + 1) + l;
        swap(a, i, r);
        return partition(a, l, r);
    

    public int partition(int[] a, int l, int r) 
        int x = a[r], i = l - 1;
        for (int j = l; j < r; ++j) 
            if (a[j] <= x) 
                swap(a, ++i, j);
            
        
        swap(a, i + 1, r);
        return i + 1;
    

    public void swap(int[] a, int i, int j) 
        int temp = a[i];
        a[i] = a[j];
        a[j] = temp;