【JavaScript 算法】堆排序：优先队列的实现

在这里插入图片描述

🔥 个人主页：空白诗

在这里插入图片描述

文章目录

- 一、算法原理
- - 堆的定义
  - 堆排序的步骤
- 二、算法实现
- - 构建最大堆
  - 注释说明：
- 三、应用场景
- 四、总结

在这里插入图片描述

堆排序（Heap Sort）是一种基于堆数据结构的排序算法，具有较好的时间复杂度表现。堆是一种特殊的完全二叉树，分为最大堆和最小堆。堆排序通过构建最大堆或最小堆来实现排序过程。本文将详细介绍堆排序算法的原理、实现及其应用。

一、算法原理

堆排序的基本思想是将待排序的数组构建成一个最大堆或最小堆，然后通过堆的删除操作将堆顶元素逐个取出，得到一个有序序列。

堆的定义

最大堆：每个节点的值都大于或等于其子节点的值。
最小堆：每个节点的值都小于或等于其子节点的值。

堆排序的步骤

构建最大堆：将数组重新组织成一个最大堆。
交换堆顶元素与末尾元素：将堆顶元素（最大值）与末尾元素交换，将最大值移到数组末尾。
调整堆：重新调整堆，保持最大堆性质。
重复步骤2和3，直到堆的大小为1，排序完成。

二、算法实现

构建最大堆

/*** 调整堆* @param {number[]} arr - 数组* @param {number} len - 堆的有效大小* @param {number} i - 当前节点的索引*/
function heapify(arr, len, i) {let largest = i; // 初始化当前节点为最大值const left = 2 * i + 1; // 左子节点索引const right = 2 * i + 2; // 右子节点索引// 如果左子节点存在且大于当前节点，则更新最大值if (left < len && arr[left] > arr[largest]) {largest = left;}// 如果右子节点存在且大于当前节点，则更新最大值if (right < len && arr[right] > arr[largest]) {largest = right;}// 如果最大值不是当前节点，则交换并继续调整堆if (largest !== i) {[arr[i], arr[largest]] = [arr[largest], arr[i]];heapify(arr, len, largest);}
}/*** 堆排序算法* @param {number[]} arr - 待排序的数组* @return {number[]} - 排序后的数组*/
function heapSort(arr) {const len = arr.length;// 构建最大堆for (let i = Math.floor(len / 2) - 1; i >= 0; i--) {heapify(arr, len, i);}// 逐个将堆顶元素移到末尾，然后调整堆for (let i = len - 1; i > 0; i--) {[arr[0], arr[i]] = [arr[i], arr[0]]; // 交换堆顶元素与末尾元素heapify(arr, i, 0); // 调整堆}return arr;
}// 示例
const arr = [3, 19, 1, 14, 8, 7];
console.log(heapSort(arr)); // 输出: [1, 3, 7, 8, 14, 19]

注释说明：

调整堆：
- heapify(arr, len, i)：调整堆，使其保持最大堆性质，接受数组、堆的有效大小和当前节点的索引作为参数。
- let largest = i;：初始化当前节点为最大值。
- const left = 2 * i + 1; const right = 2 * i + 2;：计算左、右子节点的索引。
- if (left < len && arr[left] > arr[largest]) largest = left;：如果左子节点大于当前节点，更新最大值。
- if (right < len && arr[right] > arr[largest]) largest = right;：如果右子节点大于当前节点，更新最大值。
- if (largest !== i) [arr[i], arr[largest]] = [arr[largest], arr[i]]; heapify(arr, len, largest);：如果最大值不是当前节点，则交换并继续调整堆。
堆排序：
- heapSort(arr)：堆排序算法，接受待排序的数组作为参数，返回排序后的数组。
- const len = arr.length;：获取数组长度。
- for (let i = Math.floor(len / 2) - 1; i >= 0; i--) heapify(arr, len, i);：从最后一个非叶子节点开始构建最大堆。
- for (let i = len - 1; i > 0; i--) [arr[0], arr[i]] = [arr[i], arr[0]]; heapify(arr, i, 0);：逐个将堆顶元素移到末尾，然后调整堆。