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

兩尛2025-11-20 19:33

题目描述：

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

代码思路：

java 复制代码

class Solution {
     public int findKthLargest(int[] nums, int k) {
        creatheap(nums);
        int kmax = nums[0];
        for(int i =0;i<k;i++){
            kmax = nums[0];
            nums[0] = nums[nums.length-1-i];
            heapadiust(nums,0,nums.length-2-i);
        }
        return kmax;
    }

    public void heapadiust(int[] nums,int s,int m){ //假设已经有序
        int temp = nums[s];
        int j = 2*s+1;//找到左孩子
        while(j<=m){
            if(j<m && nums[j]<nums[j+1]){ //找到左右孩子中较大的
                j++;
            }
            if(temp>=nums[j]){
                break;// 已经是大根堆了，不用调整
            }
            nums[s] = nums[j];
            s = j;
            j = 2*s+1;//找到左孩子
        }
        nums[s] = temp;
    }
    public void creatheap(int[] nums){
        for(int i=nums.length/2-1;i>=0;i--){
            heapadiust(nums,i,nums.length-1);
        }
    }
}