堆排序:力扣215.数组中的第K个大元素

server/2025/3/19 15:52:11/

一、问题描述

在一个整数数组 nums 中,需要找出第 k 个最大的元素。这里要注意,我们要找的是数组排序后的第 k 个最大元素,而不是第 k 个不同的元素。例如,对于数组 [3,2,1,5,6,4],当 k = 2 时,第 2 个最大的元素是 5。

二、解决思路

我们可以使用大顶堆排序算法来解决这个问题。大顶堆是一种完全二叉树,其中每个节点的值都大于或等于其子节点的值。通过构建大顶堆,我们可以将数组中的元素按照从大到小的顺序排列,然后取出第 k 个元素即可。

三、代码实现

javascript">/*** @param {number[]} nums* @param {number} k* @return {number}*/
var findKthLargest = function(nums, k) {let n = nums.length;// 以大顶堆为做法,其实堆就只是通过完全二叉树的性质,在数组中进行操作// 如果数组以 0 开头,那么节点 i 的左节点为 2 * i + 1,右节点为 2 * i + 2// max 表示维护的是 0 到 max 这部分的顺序,max 到 n - 1 这部分已经形成顺序了// i 表示维护的是以 i 为父节点,它和它子节点的大顶堆关系function heapify(max, i) {let lastMax = i;let lson = 2 * i + 1;let rson = 2 * i + 2;// 比较左子节点和父节点的大小,如果左子节点更大,则更新 lastMaxif (lson < max && nums[lson] > nums[lastMax]) {lastMax = lson;}// 比较右子节点和当前最大节点的大小,如果右子节点更大,则更新 lastMaxif (rson < max && nums[rson] > nums[lastMax]) {lastMax = rson;}// 如果 lastMax 不等于 i,说明需要交换父节点和最大子节点的位置if (lastMax != i) {let temp = nums[lastMax];nums[lastMax] = nums[i];nums[i] = temp;// 递归调用 heapify 函数,继续维护以 lastMax 为根节点的子树的大顶堆性质heapify(max, lastMax);}}// 大顶堆排序的入口function heap_sort() {// 首先要从最后一个非叶子节点开始建立大顶堆for (let i = Math.floor(n / 2 - 1); i >= 0; i--) {heapify(n, i);}// 然后呢,我们已经形成大顶堆了,现在要把他们完全排序// 怎么做呢,先把 [0] 与 [n - 1] 对调,这样 [n - 1] 的数就变成了最大的数// 接下来再维护 0 到 n - 2 之间的大顶堆,从堆顶 0 依次维护下去,数组 nums 就变成从小到大顺序的for (let i = n - 1; i >= 0; i--) {//这里是为了完整的堆排序,如果只是要获取第k大的//直接if(n==n-k-1) return nums[n-k]就行,无需再继续了let temp = nums[i];nums[i] = nums[0];nums[0] = temp;heapify(i, 0);}}heap_sort();console.log(nums);return nums[n - k];
};

四、代码详细解释

1. findKthLargest 函数

这个函数是整个算法的入口,它接收两个参数:nums 数组和 k。首先,我们获取数组的长度 n,然后调用 heap_sort 函数对数组进行排序,最后返回排序后数组的第 n - k 个元素,即为第 k 个最大的元素。

2. heapify 函数

这个函数用于维护以 i 为父节点的子树的大顶堆性质。具体步骤如下:

  • 初始化 lastMax 为 i,表示当前最大节点为父节点。
  • 计算左子节点的索引 lson = 2 * i + 1 和右子节点的索引 rson = 2 * i + 2
  • 比较左子节点和父节点的大小,如果左子节点更大,则更新 lastMax 为左子节点的索引。
  • 比较右子节点和当前最大节点的大小,如果右子节点更大,则更新 lastMax 为右子节点的索引。
  • 如果 lastMax 不等于 i,说明需要交换父节点和最大子节点的位置,然后递归调用 heapify 函数,继续维护以 lastMax 为根节点的子树的大顶堆性质。

3. heap_sort 函数

这个函数是大顶堆排序的入口,具体步骤如下:

  • 构建大顶堆:从最后一个非叶子节点开始,依次调用 heapify 函数,将数组转换为大顶堆。最后一个非叶子节点的索引为 Math.floor(n / 2 - 1)
  • 排序:将堆顶元素(即数组的第一个元素)与数组的最后一个元素交换,然后将堆的大小减 1,再次调用 heapify 函数维护堆的性质。重复这个过程,直到堆的大小为 1,此时数组就已经按从小到大的顺序排序好了。

http://www.ppmy.cn/server/176278.html

相关文章

m4i.22xx-x8系列-PCIe总线直流耦合5G采集卡

m4i.22xx-x8系列-PCIe总线直流耦合采集卡 四通道1.25 GS/s&#xff1b;两通道2.5GS/s&#xff1b;单通道5GS/s&#xff1b;500 MHz或1.5GHz带宽&#xff1b;标准4 GSample板载内存 概述&#xff1a; m4i.22xx-x8系列是针对高速数据采集而设计&#xff0c;包含9个型号。模块的…

Cursor AI IDE

前言 Cursor IDE 描述-->AI代码编辑器、AI IDE、集成开发环境 Cursor IDE: 这是一个集成了 AI 功能的代码编辑器&#xff0c;基于 Visual Studio Code (VSCode)&#xff0c;即带有内置 AI 的 VSCode。它利用 AI 技术来增强编程体验&#xff0c;提供智能代码补全、错误检测、…

某快餐店用户市场数据挖掘与可视化

1、必要库的载入 import pandas as pd import matplotlib.pyplot as plt import seaborn as sns2、加载并清洗数据 # 2.1 加载数据 df pd.read_csv(/home/mw/input/survey6263/mcdonalds.csv)# 2.2 数据清洗 # 2.2.1 检查缺失值 print(缺失值情况&#xff1a;) print(df.isn…

C++特性——智能指针

为什么需要智能指针 对于定义的局部变量&#xff0c;当作用域结束之后&#xff0c;就会自动回收&#xff0c;这没有什么问题。 当时用new delete的时候&#xff0c;就是动态分配对象的时候&#xff0c;如果new了一个变量&#xff0c;但却没有delete&#xff0c;这会造成内存泄…

【canvas】一键自动布局:如何让流程图节点自动找到最佳位置

一键自动布局&#xff1a;如何让流程图节点自动找到最佳位置 引言 在流程图、拓扑图和系统架构图设计中&#xff0c;节点布局往往是最令人头疼的问题。如果手动调整每个节点位置&#xff0c;不仅耗时费力&#xff0c;还难以保证美观性和一致性。本文将深入解析如何实现自动布…

flutter 专题 九十八 Flutter 1.7正式版发布

此次发布的版本是继上次 I/O大会众多重要功能发布以来的一次小更新。Flutter 1.7 包含了对 AndroidX 的支持&#xff0c;满足了 Play 商店近期对应用提出的要求&#xff0c;包含了一些新的和增强过的组件&#xff0c;修复了开发者们提出的 bug 等。 如果你已经安装并使用默认稳…

Paper Reading: AnomalyGPT:利用大型视觉-语言模型检测工业异常 (AAAI 2024 Oral)

目录 简介动机/目标相关工作基于特征嵌入的方法 feature embedding-based基于重建的方法 reconstruction-based 方法3.1 模型架构3.2 译码器和提示学习器3.3 图像-文本对齐数据异常生成 3.4 损失函数 实验比较消融 总结AppendixA. 现有IAD方法的更多实验结果B. 正常与异常文本C…

不像人做的题————十四届蓝桥杯省赛真题解析(上)A,B,C,D题解析

题目A&#xff1a;日期统计 思路分析&#xff1a; 本题的题目比较繁琐&#xff0c;我们采用暴力加DFS剪枝的方式去做&#xff0c;我们在DFS中按照8位日期的每一个位的要求进行初步剪枝找出所有的八位子串&#xff0c;但是还是会存在19月的情况&#xff0c;为此还需要在CHECK函数…