js用类实现一个最小堆数据结构

一、堆

• 插入元素
• 删除堆顶
• 获取某个元素的父元素的下标
• 获取某个父元素的左孩子节点的下标
• 获取某个父元素的右孩子节点的下标
• 获取堆的大小
• 获取堆顶

二、实现

class MinHeap {
  constructor() {
    // 堆以数组表示
    this.heap = []
  }
  // 插入操作
  insert(val) {
    this.heap.push(val)
    this.moveUp(this.heap.length - 1)
  }
  // 删除堆顶的操作
  pop() {
    this.heap[0] = this.heap.pop()
    this.moveDown(0)
  }
  // 获取堆顶元素
  top() {
    return this.heap[0]
  }
  // 获取堆的大小
  size() {
    return this.heap.length
  }
  // 获取当前元素的父元素的index
  getParentIndex(index) {
    return index - 1 >> 2
  }
  // 获取当前元素的左子元素的index
  getLeftIndex(index) {
    return index * 2 + 1
  }
  // 获取当前元素的右子元素的index
  getRightIndedx(index) {
    return index * 2 + 2
  }
  // 将当前元素与他的父元素的大小进行比较，当父元素值大于子元素时，交换两个元素，递归操作
  moveUp(index) {
    const parentIndedx = this.getParentIndex(index)
    if (this.heap[index] < this.heap[parentIndedx]) {
      this.change(index, parentIndedx)
      this.moveUp(parentIndedx)
    }
  }
  // 将当前元素与他的左右子元素的大小进行比较，当父元素值大于左子元素时，交换两个元素，递归操作；右子元素操作相同
  moveDown(index) {
    let leftIndex = this.getLeftIndex(index)
    let rightIndex = this.getRightIndedx(index)
    if (this.heap[index] > this.heap[leftIndex]) {
      this.change(index, leftIndex)
      this.moveDown(leftIndex)
    }
    if (this.heap[index] > this.heap[rightIndex]) {
      this.change(index, rightIndex)
      this.moveDown(rightIndex)
    }
  }
  // 交换两个元素的封装方法
  change(index1, index2) {
    let temp = this.heap[index1]
    this.heap[index1] = this.heap[index2]
    this.heap[index2] = temp
  }
}

三、应用

• 求TOP K 求数组的最大的前K个元素，用大顶堆；求数组的最小的前K个元素用小顶堆