LeetCode 215: Kth Largest Element in an Array

最新推荐文章于 2024-11-19 02:25:22 发布

weixin_33895475

最新推荐文章于 2024-11-19 02:25:22 发布

阅读量47

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/shuashuashua/p/7418244.html

本文介绍了一种寻找数组中第K大的元素的算法实现，采用快速选择算法思想，通过分区减少递归调用的范围，提高了查找效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

class Solution {
    public int findKthLargest(int[] nums, int k) {
        if (nums.length == 0) {
            return 0;
        }
        return findKth(nums, 0, nums.length - 1, nums.length - k);
    }
    
    private int findKth(int[] nums, int start, int end, int k) {
        if (start > end) {
            return -1;
        }
        int pivot = nums[end];
        int runner = start;
        for (int i = start; i < end; i++) {
            if (nums[i] <= pivot) {
                swap(nums, i, runner++);
            }
        }
        
        swap(nums, runner, end);
        
        if (runner == k) {
            return nums[runner];
        } else if (runner < k) {
            return findKth(nums, runner + 1, end, k);
        }
        return findKth(nums, start, runner - 1, k);
    }
    
    private void swap(int[] nums, int i, int j) {
        int tmp = nums[i];
        nums[i] = nums[j];
        nums[j] = tmp;
    }
}