215、数组中的第K个最大元素 | 堆应用-JS

给定整数数组 nums 和整数 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

提示：

1 <= k <= nums.length <= 104
-104 <= nums[i] <= 104

class MinHeap {
  constructor(){
    this.heap = [];
  }
  swap(i1, i2){
    const temp = this.heap[i1];
    this.heap[i1] = this.heap[i2];
    this.heap[i2] = temp;
  }
  getParentIndex(i){  //获取父节点的值
    return (i-1) >> 1;  //二进制右移相当于除以2
  }
  getLeftIndex(i) {  //获取左结点
    return i * 2 + 1;
  }
  getRightIndex(i) {  //获取右结点
    return i * 2 + 2;
  }

  shiftUp(index){   //需要让父节点不断小于它的子节点
    if(index == 0){ return; }  //如果已经是根结点了就不用找了
    const parentIndex = this.getParentIndex(index);
    if(this.heap[parentIndex] > this.heap[index]){
      this.swap(parentIndex, index);  //如果父节点的值大于子节点则进行交换
      this.shiftUp(parentIndex)
    }
  }
  insert(value){  //插入，添加的方法
    this.heap.push(value);
    this.shiftUp(this.heap.length - 1);  //shiftUp就是上移操作，接收参数是上移时的下标
  }
  shiftDown(index){  //下移节点，直到子节点都大于当前节点的值
    // 需要获取它的左右子节点
    const leftIndex = this.getLeftIndex(index);
    const rightIndex = this.getRightIndex(index);
    if(this.heap[leftIndex] < this.heap[index]){  //小顶堆，父节点小于子节点
      this.swap(leftIndex, index);
      this.shiftDown(leftIndex);  //递归，直到找到合适的位置
    }
    if(this.heap[rightIndex] < this.heap[index]){  //小顶堆，父节点小于子节点
      this.swap(rightIndex, index);
      this.shiftDown(rightIndex);  //递归，直到找到合适的位置
    }
  }

  pop(){   //下移方法
    this.heap[0] = this.heap.pop();  // 把数组的最后一位转移到头部，相当于变相删除堆顶
    this.shiftDown(0);  //传什么下标，就对那个进行下移操作
  }
  peek(){ //获取堆顶，返回数组的头部
    return this.heap[0];
  }
  size(){  // 获取堆的大小
    return this.heap.length;
  }
}
/**
 * @param {number[]} nums
 * @param {number} k
 * @return {number}
 */
var findKthLargest = function(nums, k) {
    const h = new MinHeap(); //构建一个最小堆
    nums.forEach(n => {
        h.insert(n);
        if(h.size() > k) {  //大于堆的个数，就把堆顶删除
           h.pop();
        }
    })
    return h.peek();  //所以都插入完成之后，堆顶就是要的元素
};