数组中第 k 个最大的元素

//给定整数数组 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 
//
// 
//
// 提示： 
//
// 
// 1 <= k <= nums.length <= 10⁵ 
// -10⁴ <= nums[i] <= 10⁴ 
// 
//
// Related Topics 数组 分治 快速选择 排序 堆（优先队列） 👍 2465 👎 0


//leetcode submit region begin(Prohibit modification and deletion)
class Solution {
    public int findKthLargest(int[] nums, int k) {
        int heapSize = nums.length;
        buildHeap(nums, heapSize);
        for (int i = nums.length - 1; i >= nums.length - k + 1; i--) {
            swap(nums, 0,  i);
            heapSize--;
            adjustHeap(nums, 0, heapSize);
        }
        return nums[0];
    }

    /**
     * 建堆
     * @param nums
     * @param heapSize
     */
    /**
     * 建堆
     * @param nums
     * @param heapSize
     */
    private void buildHeap(int[] nums, int heapSize){
        for (int i = heapSize/2 -1; i >= 0 ; i--) {
            adjustHeap(nums, i, heapSize);
        }
    }

    /**
     * 堆调整
     * @param nums
     * @param heapSize
     */
    private void adjustHeap(int[] nums, int index, int heapSize){
        int left = 2*index +1, right = index *2 +2;
        if (index < heapSize){
            int max = index;
            if(left < heapSize && nums[left] > nums[max]){
                max = left;
            }
            if(right < heapSize && nums[right] > nums[max]){
                max = right;
            }
            if(max != index){
                swap(nums, max, index);
                adjustHeap(nums, max, heapSize);
            }
        }
    }


    /**
     * 交换
     * @param nums
     * @param i
     * @param j
     */
    private void swap(int[] nums, int i, int j){
        int temp = nums[i];
        nums[i] = nums[j];
        nums[j] = temp;
    }
}
//leetcode submit region end(Prohibit modification and deletion)