再谈Heap(堆)的妙用
好久没更新了,今天我又来了。一直看我文章的朋友可能会发现,我之前多次提到堆这个数据结构,它可以帮助我们快速地在一堆数据中,找到符合某个条件的最大或者最小的元素。今天我们还是要来谈论它,没办法,它就是这么强大又好用。
来看这样一道题目:给定K个排序好的链表,排序好合并到一个列表中去。
示例 1:
输入: L1=[2, 6, 8], L2=[3, 6, 7], L3=[1, 3, 4]
输出: [1, 2, 3, 3, 4, 6, 6, 7, 8]
示例 2:
输入: L1=[5, 8, 9], L2=[1, 7]
输出: [1, 5, 7, 8, 9]
拿到这道题的第一感觉似乎也不是很难嘛,我们可以把所有元素加到一个列表里面去然后再排序,时间复杂度也不过是O(N∗logN)。这里N是K个链表所有元素之和。
不过这么做的话,其实链表们排没排序都一样,既然题目告诉我们链表是排序好的,那它肯定是更优解法的重要条件。我们得想办法利用好这个条件。我们想要让最终的列表排好序,那肯定要把所有元素里最小的元素放在前面。既然给我们的K个链表都是排好序的,那我们只要比较它们的第一个元素就能得到最终列表的最小元素里。拿到最小元素我们就添加到最终列表里,按照这个思路,我们在每一步都可以拿到K个链表中的最小元素,最后得到我们想要的结果。
那说到在K个元素中找最小的元素,我们又想到堆了。那我们再来盘一下用堆我们该怎么做:
- 把每个链表第一个元素插入到最小堆
- 从堆中取出最小的元素添加到结果列表中
- 再从拿出去的元素所在的那个链表中取出下一个元素放到堆中
- 重复第2步跟第3步,我们可以保证所有元素添加到了结果列表中且有序
就拿第一个例子来说,我们再来详细讨论一下这个过程。
给定的链表: L1=[2, 6, 8],
L2=[3, 6, 7],
L3=[1, 3, 4]
- 在从每个链表中取出第一个元素时,我们的堆是这样的:
- 我们会取出位于堆顶的元素,加入到合并的列表中,然后把这个元素所在链表中的的下一个元素加入到堆中:
- 那么,我们再次把堆顶的元素取出放到结果列表中,把下一个元素添加到堆中:
- 重复上面的步骤,取出堆顶元素加入到结果列表中,再把它的下一个元素加入到堆中。这里有两个3,可以随便选,但是要注意选了哪个就把哪个的下一个元素加到堆中,这个次序很重要。
最终我们就能得到一个排序好的列表啦!
public static ListNode merge(ListNode[] lists) {
PriorityQueue<ListNode> minHeap = new PriorityQueue<ListNode>((n1, n2) -> n1.value - n2.value);
// 把第一个元素添加到堆
for (ListNode root : lists)
if (root != null)
minHeap.add(root);
// 从最小堆中取出最小的元素并加入到结果中
// 如果它在链表中还有下一个元素,那么把它的这个下一个元素添加到堆中
ListNode resultHead = null, resultTail = null;
while (!minHeap.isEmpty()) {
ListNode node = minHeap.poll();
if (resultHead == null) {
resultHead = resultTail = node;
} else {
resultTail.next = node;
resultTail = resultTail.next;
}
if (node.next != null)
minHeap.add(node.next);
}
return resultHead;
}
看看,这个问题有了堆这个数据结构变得格外简单,复杂度也得到了优化!这里时间复杂度在O(N∗logK),而空间复杂度在O(K)。这就是我们常听到的多路归并算法的核心思想,以后再看到类似题目知道该怎么办了吧?
最后的最后,我再强调下复习一下常见的数据结构的重要性,会对我们解题大有帮助。就拿这道题来说,思路其实很简单,但是如果不知道或者忘了堆的用法,这道题大概率只能想到把所有元素加入结果列表再排序,岂不可惜?
- awk中NF的使用
- tar.gz 解压
- Python&机器学习之项目实践
- JAVA CDI 学习(5) - 如何向RESTFul Service中注入EJB实例
- mysql5.7 column cannot be null
- 区块链大热 价值近20万的Matrix.io被启用
- 比特币科普之什么是区块高度?
- 如何正确并快速理解MapReduce
- mysqldump的简单使用
- mac:在当前文件夹打开terminal终端
- 斯诺登研发黑客警报App上架 监测功能引发争议
- JMS + jboss EAP 6.2 示例
- Maven Compilation error [package org.testng.annotations does not exist]
- 查看jar包的jdk版本并降级
- JavaScript 教程
- JavaScript 编辑工具
- JavaScript 与HTML
- JavaScript 与Java
- JavaScript 数据结构
- JavaScript 基本数据类型
- JavaScript 特殊数据类型
- JavaScript 运算符
- JavaScript typeof 运算符
- JavaScript 表达式
- JavaScript 类型转换
- JavaScript 基本语法
- JavaScript 注释
- Javascript 基本处理流程
- Javascript 选择结构
- Javascript if 语句
- Javascript if 语句的嵌套
- Javascript switch 语句
- Javascript 循环结构
- Javascript 循环结构实例
- Javascript 跳转语句
- Javascript 控制语句总结
- Javascript 函数介绍
- Javascript 函数的定义
- Javascript 函数调用
- Javascript 几种特殊的函数
- JavaScript 内置函数简介
- Javascript eval() 函数
- Javascript isFinite() 函数
- Javascript isNaN() 函数
- parseInt() 与 parseFloat()
- escape() 与 unescape()
- Javascript 字符串介绍
- Javascript length属性
- javascript 字符串函数
- Javascript 日期对象简介
- Javascript 日期对象用途
- Date 对象属性和方法
- Javascript 数组是什么
- Javascript 创建数组
- Javascript 数组赋值与取值
- Javascript 数组属性和方法
- 机器学习笔记之梯度下降算法原理讲解
- 使用OpenCV在Python中进行图像处理
- R语言对布丰投针(蒲丰投针)实验进行模拟和动态可视化生成GIF动画
- Shiny-R语言轻松开发交互式web应用
- SPSS中的等级线性模型Multilevel linear models研究整容手术数据
- Python花式表白
- Python中基于网格搜索算法优化的深度学习模型分析糖尿病数据
- WordPress运行内存不足的解决方式
- ggplot2 |legend参数设置,图形精雕细琢
- miRBase物种名缩写
- Python中基于网格搜索算法优化的深度学习模型分析糖尿病数据
- 在R语言中使用航空公司复杂网络对疫情进行建模
- R语言用关联规则和聚类模型挖掘处方数据探索药物配伍中的规律
- ggplot2|theme主题设置,详解绘图优化-“精雕细琢”
- ggalluvial|炫酷桑基图(Sankey),你也可以秀