js排序 - 堆排序

堆排序

堆排序是指利用堆这种数据结构所设计的一种排序算法。堆是一个近似完全二叉树的结构，并同时满足堆积的性质：即父节点的值总是大于其子节点的值，即“大 顶堆”(或反之，称为“小顶堆”）。

工作原理

将待排序的数组构造成一个大顶堆。此时整个数组的最大值就是堆顶的根节点。将它移走，其实就是将其与堆数组的末尾元素交换，此时末尾元素就是最大值。然后将剩余的 n-1 个元素又重新构造成堆，这样就又能得到次大值。
如此反复操作，直到只剩余一个元素，就能得到一个有序数组了。

图解

重复上图过程，直到整个数组都成为「大顶堆」结构为止
注意：每一个非叶子节点与它的两个子节点的索引值关系：假设这个非叶子节点的索引为 index
root: index
left: 2 * index + 1
right: 2 * index + 2

js代码实现 + 效率测试

运算10次，平均耗时15.38 ms

// 创建 20000 个随机数，数值范围：1 - 100000
let ary = [];
for (let i = 0; i < 20000; i++) {
  ary.push( Math.floor( Math.random()*99999 + 1 ) );
}

// 堆排序
function heapSort(arr) {
  // 第一次初始化大顶堆
  let len = arr.length - 1;
  let lastLeaf = Math.floor( (len + 1) / 2 ) - 1;
  for (let i = lastLeaf; i >= 0; i--) {
    buildHeap(arr, i, len);
  }
  swap(arr, 0, len);
  len--;

  // 因为第一次已经把数组初始化为大顶堆了，第一次「交换根节点与最后一个位置」之后，只有根节点的位置是乱的
  // 所以只要从根节点向下遍历即可，把他与较大的那个子节点交换位置
  // 直到遍历完未排序的元素或者没有比他再大的子节点为止
  for (let k = len; k > 0; k--) {
    buildHeap(arr, 0, k);
    swap(arr, 0, k);
  }
}

// 构建大顶堆
function buildHeap(arr, k, len) {
  for (let i = k; i <= len;) {
    let l = 2 * i + 1, r = 2 * i + 2, next = i;
    if (l <= len && arr[l] > arr[next]) {
      next = l;
    }
    if (r <= len && arr[r] > arr[next] ) {
      next = r;
    }
    if (i !== next) {
      swap(arr, i, next);
      i = next;
    }
    else {
      break;
    }
  }
}

// 交换元素位置
function swap(arr, l, r) {
  let temp = arr[l];
  arr[l] = arr[r];
  arr[r] = temp;
}

console.time( 'heapSort' );
heapSort(ary);
console.timeEnd( 'heapSort' );

说明

本人也是第一次画图，图画的不好，大佬们多多提意见，我会改正😋，建议新手朋友可以动手画一下流程，这样就好理解了

建议

建议优先掌握「快速排序」「堆排序」，这两种排序算法相对来说使用场景比较多，算法思想非常好，但也不是说其他的算法不重要哦~