LintCode 606.第K大的元素Ⅱ

最新推荐文章于 2021-06-07 16:05:02 发布

零一熊

最新推荐文章于 2021-06-07 16:05:02 发布

阅读量69

点赞数

分类专栏： LintCode 文章标签： leetcode java

本文链接：https://blog.csdn.net/weixin_50899769/article/details/116330588

版权

LintCode 专栏收录该内容

51 篇文章 0 订阅

订阅专栏

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

Partition
随机选取一个数组元素pivot 将数组分成两部分左边小于pivot 右边大于pivot
看k落在哪段区间上对这段区间继续partition
直到pivot的位置与k相同返回pivot

public class Solution {
    /**
     * @param nums: an integer unsorted array
     * @param k: an integer from 1 to n
     * @return: the kth largest element
     */
    public int kthLargestElement2(int[] nums, int k) {
        if(nums == null || nums.length < k || nums.length == 0)
            return -1;
        return helper(nums.length - k + 1, nums, 0, nums.length - 1);
    }
    private int helper(int k, int[] nums, int l, int r){
        if(l == r)
            return nums[l];
        int position = partition(nums, l, r);
        if(position + 1 == k)
            return nums[position];
        else if(position + 1 > k)
            return  helper(k, nums, l, position - 1);
        else
            return helper(k, nums, position + 1, r);
    }
    private int partition(int[] nums, int l, int r){
        int left = l, right = r;
        int pivot = nums[left];
        while(left < right){
            while(left < right && nums[right] >= pivot){
                right--;
            }
            nums[left] = nums[right];
            while(left < right && nums[left] <= pivot){
                left++;
            }
            nums[right] = nums[left];
        }
        nums[left] = pivot;
        return left;
    }
}