今天的两道题目全都围绕链表,第一个是困难级别的、要合并多个排序的链表;第二题是中等难度,需要两两交换链表中的节点,昨天没能用递归法写出代码,今天就尝试用递归实现了下,测试效果不咋地,但递归法跑通了!
题目一
第 23 题:合并K个排序链表:
合并 k 个排序链表,返回合并后的排序链表。请分析和描述算法的复杂度。
示例:
输入:
[
1->4->5,
1->3->4,
2->6
]
输出: 1->1->2->3->4->4->5->6
思路
刚看到这题面,联想昨天刚写了合并两个排序链表的题目,最先想到的就是基于能够实现合并两个的方法 mergeTwoLists(l1,l2),先取列表中前两条链表合并,再将结果分别与第三、四等一直到最后一条链表合并,最终结果即所求。但按这个思路写完代码提交后,测评结果是“超出时间限制”。
如果想简化处理,我们从结果来看,其实就是把所有可能的数以链表的形式输出。那么思路就来了:我们把列表中所有链表中出现的节点数字存起来,排序后用这些数字重新生成一条链表,就达到题目要求了。
代码
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def mergeKLists(self, lists: List[ListNode]) -> ListNode:
# 用来存所有数字的列表
record = []
# 遍历链表所在列表
for item in lists:
# while 循环对每条链表取值
while item!=None:
# 将链表中每个节点的当前值存入列表中
record.append(item.val)
# 移动链表中的节点
item = item.next
# 将取到的结果进行排序
record.sort()
# 建个初识节点
start = ListNode(0)
# 复制下用于最后返回
start_copy = start
# 对记录的数字进行遍历
for i in record:
# 新生成的链表每个节点值即排序的数字
start.next = ListNode(i)
# 移动生成的链表位
start = start.next
# 通过复制的初识节点的下一位来返回完整链表
return start_copy.next
感觉怎么从链表角度来考虑问题,但又确实用到了链表的 val 和 next 属性,也凑活吧。
提交
效果还不错,可能是取巧了:
执行用时 : 84 ms, 在所有 Python3 提交中击败了 86.09% 的用户
内存消耗 : 17.4 MB, 在所有 Python3 提交中击败了 7.14% 的用户
学习
翻看其它题解时,多数都是称我们刚这思路为暴力法。同时,我最初基于合并两个链表来实现的思路,被称为分治法。
字面上的解释是“分而治之”,就是把一个复杂的问题分成两个或更多的相同或相似的子问题,再把子问题分成更小的子问题……直到最后子问题可以简单的直接求解,原问题的解即子问题的解的合并。这个技巧是很多高效算法的基础,如排序算法(快速排序,归并排序),傅立叶变换(快速傅立叶变换)……
我把代码贴出来分析下: