Skip to content

Latest commit

 

History

History
54 lines (50 loc) · 1.37 KB

TheRocket.md

File metadata and controls

54 lines (50 loc) · 1.37 KB
class Solution {
    public int findKthLargest(int[] nums, int k) {
        k = nums.length - k;
        return quickSelect(nums, k);
    }

    private int quickSelect(int[] nums, int k) {
        int lo = 0;
        int hi = nums.length - 1;
        while (lo < hi) {
            int i = partition(nums, lo, hi);
            if (i < k) {
                lo = i + 1;
            } else if (i > k) {
                hi = i - 1;
            } else {
                return nums[i];
            }
        }
        return nums[lo];
    }

    private int partition(int[] nums, int lo, int hi) {
        int m = median3(nums, lo, lo + (hi - lo) / 2, hi);
        swap(nums, lo, m);
        int i = lo;
        int j = hi + 1;
        int v = nums[lo];
        while (true) {
            while (nums[++i] < v && i < hi) {}
            while (nums[--j] > v) {}
            if (i >= j) {
                break;
            }
            swap(nums, i, j);
        }
        swap(nums, lo, j);
        return j;
    }

    private static int median3(int[] nums, int i, int j, int k) {
        return nums[i] < nums[j] ?
                (nums[j] < nums[k] ? j : nums[i] < nums[k] ? k : i) :
                (nums[k] < nums[j] ? j : nums[k] < nums[i] ? k : i);
    }

    private void swap(int[] nums, int i, int j) {
        int tmp = nums[i];
        nums[i] = nums[j];
        nums[j] = tmp;
    }
}