数组中的第K个最大元素

https://leetcode-cn.com/problems/kth-largest-element-in-an-array/

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

package factory;

/**
 * @Description: 数组中的第K个最大元素
 * https://leetcode-cn.com/explore/interview/card/top-interview-quesitons-in-2018/266/heap-stack-queue/1154/
 * @Author: Jaryn
 * @Date: 2019/12/9 10:02
 */
public class FindKthLargest {

    public static void main(String[] args) {
        int[] nums = new int[]{3,2,3,1,2,4,5,5,6};
        System.out.println(findKthLargest(nums, 4));
    }

    public static int findKthLargest(int[] nums, int k) {
        //初始化树
        initTree(nums, k);
        // 开始循环
        for(int i=k ; i<nums.length; i++) {
            if(nums[i] > nums[0]) {
                swap(nums, 0, i);
                adjust(nums, k, 0);
            }
        }

        return nums[0];
    }

    public static void initTree(int[] nums, int k) {
        //对k大的树所有的父节点进行调节
        for(int i=(k-1)/2; i>=0; i--) {
            adjust(nums, k, i);
        }
    }

    public static void adjust(int[] nums, int k, int index) {
       if(index > k) {
           return;
       }

       int len = nums.length;
       int left = 2*index +1;
       int right = 2*index +2;
       int min = index;

       if(left < k && nums[left] < nums[min]) {
           min = left;
       }
       if(right < k && nums[right] < nums[min]) {
           min = right;
       }
       if(min != index) {
           swap(nums, index, min);
           adjust(nums, k, min);
       }

    }

    public static void swap(int[] nums, int x, int y) {
        int temp = nums[x];
        nums[x] = nums[y];
        nums[y] = temp;
    }


}