python链表leetcode问题21合并两个排序列表

Question

我有一个关于 python 中链表的快速问题。 在下面显示的解决方案代码中，当我尝试合并两个已排序的链表时。 我对包含的 if 和 elif 语句的条件感到困惑。 例如，如果 l1 不为空而 l2 为空，我想将 l1 中的其余 3 个元素添加到我的新链表中，但代码显示 l1 和 tail 未更新，所以它不只是添加一个3？

我的另一个问题是关于返回 head.next。 返回是否会自动将每个节点从 head.next 返回到 null ptr？ 谢谢！

# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def mergeTwoLists(self, l1: ListNode, l2: ListNode) -> ListNode:
        head = ListNode()
        tail = head
        
        while l1 and l2:
            if l1.val < l2.val:
                tail.next = l1
                l1 = l1.next
            else:
                tail.next = l2
                l2 = l2.next
            tail = tail.next
        
        if l1:
            tail.next = l1
            #why don't I update l1 and tail
        elif l2:
            tail.next = l2
            #why don't I update l2and and tail
        return head.next
        #does returning head.next return every single value from head.next to null?

Answer 1

好吧，您正在使用链表，因此如果您指向一个特定节点并且该节点在其下一个节点中有更多节点，那么您将获得所有节点。

那么这里有什么问题呢？

嗯，其实没什么。 你正在返回head -> next ，所以基本上你正在返回整个链表。 如果您像这样遍历列表：

merged_list = solution.mergeTwoLists(lst1, lst2)
while merged_list:
    print(str(merged_list.val), end = ' -> ')
    merged_list = merged_list.next
print(merged_list)

例如，如果您有以下链表lst1和lst2定义如下：

lst1 = ListNode(1)
lst1.next = ListNode(6)
lst1.next.next = ListNode(3)    # So lst1 is basically 1 -> 6 -> 3 -> None

lst2 = ListNode(4)
lst2.next = ListNode(5)
lst2.next.next = ListNode(2)    # And lst2 is basically 4 -> 5 -> 2 -> None

然后你会得到最终结果：

1 -> 4 -> 5 -> 2 -> 6 -> 3 -> None

这正是您在代码中应用的内容。