为什么链表的最后一个元素没有出现在输出中？

Question

我写了一些与链表相关的代码，ListNode 类。代码的目的是将两个有序链表合并成一个大的有序链表。

所以，例如

输入：L1 = [1,2,4]，L2 = [1,3,4]

输出：[1,1,2,3,4,4]

def mergeTwoLists(self, l1: ListNode, l2: ListNode) -> ListNode:
        
        prehead = ListNode(-1)
        if l1 and l2 == None:
            return None

        if l1 == None:
            return l2

        if l2 == None:
            return l1

        prev = prehead
        curr1 = l1
        curr2 = l2

        while curr1 and curr2:
            if curr1.val <= curr2.val:
                prev.next = curr1
                curr1 = curr1.next 

            elif curr1.val >= curr2.val:
                prev.next = curr2
                curr2 = curr2.next
                
            prev = prev.next
                

        return prehead.next

上面的代码返回 [1,1,2,3,4] 作为输出而不是正确答案 [1,1,2,3,4,4]，我知道为什么；这是因为当我们到达 L1 的最后一个元素时，while 循环停止并且没有考虑 L2 的最后一个元素。

我尝试更改代码，特别是将“while curr1 and curr2”行改为“while L1 and L2”：....

但是这为以下行产生了一个 AttributeError "if curr1.val < curr2.val:" NoneType object has no attribute 'val'。

任何想法 1. 如何修改代码以产生正确的输出 2. 为什么我在更改 while 行时会收到该错误？

谢谢

Answer 1

问题实际上来自while，因为当两个列表中的一个不再有元素时，使用“and”，因此我们必须使用“or”并处理列表为空的情况：

while curr1 or curr2:
    if curr1 and (not curr2 or curr1.val <= curr2.val):
        prev.next = curr1
        curr1 = curr1.next 

    elif not curr1 or curr1.val >= curr2.val:
        prev.next = curr2
        curr2 = curr2.next
                
    prev = prev.next