LeetCode 刷起来

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

猜你喜欢

转载自blog.csdn.net/u011228094/article/details/80498046
今日推荐