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

数组中的第k个最大元素

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

示例 1:

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

输出: 5

示例 2:

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

输出: 4

说明:

你可以假设 k 总是有效的，且 1 ≤ k ≤ 数组的长度。

 

class Solution {
    public int findKthLargest(int[] nums, int k) {
        int begin=0;
        int end=nums.length-1;
        k=nums.length+1-k;
        while(begin<end){
            int pos=partition(nums,begin,end);
            if(pos==k-1) break;
            else if(pos<k-1) begin=pos+1;
            else end=pos-1;
        }
        return nums[k-1];
    }
    public int partition(int[] nums, int low, int high) {//分区
        int key;
        if (low > high) {
            return 0;
        }
        key = nums[low]; // 用子表的第一个记录做基准
        while (low < high) { // 从表的两端交替向中间扫描
            while (low < high && nums[high] >= key)
                high--;
            if (low < high)
                swap(nums,low,high);// 用比基准小的记录替换低位记录
            while (low < high && nums[low] < key)
                low++;
            if (low < high) // 用比基准大的记录替换高位记录
                swap(nums,low,high);
        }
        return low;
    }

    private void swap(int[] a, int i, int j) {
        int t = a[i];
        a[i] = a[j];
        a[j] = t;
    }
}