1. Merge Two Sorted Lists (0529)
合并两个排序好的链表。
Example:
Input: 1->2->4, 1->3->4
Output: 1->1->2->3->4->4
方法:遍历两个链表,每次取最小值。
思路:
我们不妨换个角度来理解题目意思:将一系列有序节点插入到链表中,使得链表仍保持有序。
想想插入一个节点的时候是怎么样的:顺序遍历链表,直到在链表中找到第一个大于此节点值的节点,将此节点插入到该节点前。现在来考虑插入一系列节点的时候,我们当然可以对每个节点都重复上述算法,即对每个节点都遍历一次链表找到要插入的位置并插入,但这么做有没有优化的空间呢?
注意到这一系列要插入的节点是有序的,那么也就是说这一系列节点中,后一个节点的插入位置一定在前一个节点的插入位置之后(因为插入后的整个链表是有序的);也就是说,我们每次在插入新的节点时,都可以以上一个节点的插入位置作为起点,去寻找插入位置并插入。
如上所述是本题算法的主要部分,下面我们来看边界情况。首先看插入第一个节点时,要插入的第一个节点可能小于链表头,也可能大于链表头,大于链表头时很容易处理,我们主要关注小于链表头的情况,因为在链表的算法中更改链表头是一件很麻烦的事情,该怎么解决呢?很简单,再回到题目,既然题目没有规定两个有序链表哪个作插入方、哪个作被插入方,那不妨指定链表头较小的那个链表左被插入方,插入方(另一链表)的第一个节点(链表头)一定插入到链表头之后。
再看另一边界情况:节点全部插入完而链表未遍历完,或链表遍历完而节点还未插入完。先看第一种,既然节点全都插入到链表中了,那不妨就直接返回了,原链表也没有需要特殊处理的了。再看第二种,当在寻找节点插入的位置时,遍历完了链表,那就说明此时剩下的节点全都比链表最大值还要大,既然剩下的节点都是有序且连在一起的,那不妨按顺序全都接在链表尾就好了。
把上面的思路串起来这题也就解决了。下面的代码用了一些编程技巧,看不太懂的话找个例子跟着走一遍就明白啦。
代码:
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def mergeTwoLists(self, l1, l2):
"""
:type l1: ListNode
:type l2: ListNode
:rtype: ListNode
"""
if l1 is None:
return l2
if l2 is None:
return l1
if l1.val > l2.val:
l1, l2 = l2, l1
p = l1
while p is not None:
while l2 is not None and (p.next is None or l2.val < p.next.val):
tmp = l2
l2 = l2.next
tmp.next = p.next
p.next = tmp
p = p.next
p = p.next
return l1