讲师中心 微信公众号
AI工具推荐 视频效率加速

第K小元素查找优化实战:利用堆结构快速提取目标变量

风芳大大_5925

风芳大大_5925

发布时间:2026-05-12 18:57:47

|

525人浏览过

|

来源于php中文网

原创

用大小为K的大根堆求第K小元素:堆顶即当前最小K个数中的最大值;新元素x小于堆顶则替换堆顶;需校验k有效性;Python中可用负值模拟大根堆。

直接用堆找第 k 小元素,关键不是“建完堆再取”,而是边遍历边维护一个大小为 k 的大根堆——堆顶始终是当前已见元素中第 k 小的候选者。

为什么选大根堆?

目标是“第 K 小”,意味着我们只关心最小的 K 个数里最大的那个(即第 K 小)。大根堆天然能快速访问这 K 个数里的最大值:

  • 堆里最多存 K 个元素,且始终保持堆顶 ≥ 堆内其余所有元素
  • 新来一个数 x:若 x
  • 遍历结束后,堆顶就是全局第 K 小元素

操作步骤要精简

不需要预建堆或排序整个数组,按顺序扫描原数据即可:

  • 初始化空的大根堆(如 C++ 的 priority_queue<int>,Java 的 PriorityQueue<Integer> 默认小根堆,需传入 Comparator.reverseOrder())
  • 对每个元素 nums[i]:
    • 若堆 size < K:直接 push
    • 若堆 size == K 且 nums[i] < 堆顶:pop 堆顶,push nums[i]
    • 否则跳过
  • 循环结束,返回堆顶

时间与空间开销很实际

适合在线场景或数据流处理:

  • 时间复杂度:O(n log k),每轮最多一次 log k 的堆调整,比全排序 O(n log n) 更优,尤其当 k ≪ n 时
  • 空间复杂度:O(k),只存 K 个关键值,内存友好
  • 支持动态插入:后续新增元素可复用同一逻辑,无需重建堆

注意边界和实现细节

几个容易出错但影响结果的地方:

  • K 超出范围(≤ 0 或 > 数组长度)需提前校验并返回错误值或抛异常
  • 语言差异:Python 的 heapq 只提供小根堆,要模拟大根堆,可存负值或用 heapq._heapify_max(非公开API,不推荐),更稳妥是用 heapq 维护小根堆来求第 K 大,或改用 sortedcontainers 等第三方库
  • 重复元素:算法天然兼容重复值,比如 [1,1,2,2,3], k=3 → 第 3 小是 2,逻辑完全正确

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

热门AI工具

更多
SkildArt
SkildArt Hot

SkildArt是一款AI文本写作工具,一站式 AI 视觉创作平台。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

墨刀AI
墨刀AI Hot

一款AI图像与设计工具,主要用于产品经理的专属智能体,适合需要提升相关任务效率的用户。

蛙蛙写作

一款AI论文写作工具,主要用于超级AI智能写作助手,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

音述AI
音述AI Hot

一款AI音频处理工具,主要用于音述AI是一个以“用声音述说故事”为核心的 AI 音乐创作与声音分享社区,适合需要提升相关任务效率的用户。

二狗PPT
二狗PPT Hot

一款AI演示文稿工具,主要用于专为中式职场打造的AI PPT生成工具,适合需要提升相关任务效率的用户。

超级简历WonderCV

一款AI办公效率工具,主要用于免费求职简历模版下载制作,应届生职场人必备简历制作神器,适合需要提升相关任务效率的用户。

相关专题

更多
堆和栈的区别
堆和栈的区别

堆和栈的区别:1、内存分配方式不同;2、大小不同;3、数据访问方式不同;4、数据的生命周期。本专题为大家提供堆和栈的区别的相关的文章、下载、课程内容,供大家免费下载体验。

5347

2023.07.18

堆和栈区别
堆和栈区别

堆(Heap)和栈(Stack)是计算机中两种常见的内存分配机制。它们在内存管理的方式、分配方式以及使用场景上有很大的区别。本文将详细介绍堆和栈的特点、区别以及各自的使用场景。php中文网给大家带来了相关的教程以及文章欢迎大家前来学习阅读。

2348

2023.08.10

页面置换算法
页面置换算法

页面置换算法是操作系统中用来决定在内存中哪些页面应该被换出以便为新的页面提供空间的算法。本专题为大家提供页面置换算法的相关文章,大家可以免费体验。

5476

2023.08.14

Kratos框架HTTP与gRPC服务开发教程
Kratos框架HTTP与gRPC服务开发教程

本专题围绕Kratos框架双协议服务开发,涵盖HTTP路由与处理器编写、参数获取、gRPC服务实现与客户端调用、metadata上下文传递、encoding编解码注册、统一响应封装、超时控制与流式响应实现方法。

0

2026.10.10

Kratos框架Protobuf接口定义与代码生成合集
Kratos框架Protobuf接口定义与代码生成合集

本专题讲解Kratos框架接口定义体系,涵盖proto编写规范、proto add/client/server生成命令、http注解路由、validate校验、OpenAPI文档生成、跨服务proto复用与兼容性设计。

0

2026.10.10

C++虚函数怎么定义和调用
C++虚函数怎么定义和调用

C++虚函数是实现运行时多态的重要机制。本专题从virtual关键字的基本用法入手,介绍基类与派生类之间的函数重写、基类指针调用派生类方法,以及动态绑定的执行过程,帮助初学者掌握虚函数的核心语法。

0

2026.10.10

C++类与对象的封装方法教程
C++类与对象的封装方法教程

C++封装是面向对象编程的核心特性之一,通过类将数据与操作数据的函数组织在一起,并利用访问权限控制外部访问。本专题介绍类的定义、成员变量、成员函数以及public、private和protected的使用方法,帮助初学者掌握封装的基本原理。

0

2026.10.10

C++构造函数定义与调用方法
C++构造函数定义与调用方法

C++构造函数用于初始化类对象,是面向对象编程的重要基础。本专题从构造函数的定义、声明和调用入手,介绍默认构造函数、带参数构造函数、拷贝构造函数及成员初始化列表,帮助初学者掌握对象创建与初始化的基本方法。

0

2026.10.10

Kratos框架零基础入门教程
Kratos框架零基础入门教程

本专题整理Kratos框架入门内容,涵盖Go环境准备、kratos CLI安装升级、new命令创建项目、目录结构分层说明、服务启动与双协议端口、依赖下载报错排查,帮助开发者快速跑通第一个Kratos框架微服务应用。

0

2026.10.10

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn