错误：在 Python 中使用虚拟头技术合并两个排序的链表

Question

我正在尝试使用虚拟头技术解决合并两个排序的链表问题。出于某种原因，我将参数传递给我的假头持有人时出错。输出假设像这样合并两个链表：1-> 1-> 2-> 3-> 7-> None

如果您能指导我的虚拟头变量中需要传递哪些数据，我会很高兴？提前致谢！这是我遇到的错误：

    dummy = LinkedList()
TypeError: __init__() missing 1 required positional argument: 'data

这是我的完整代码：

class LinkedList:
    def __init__(self, data):
        self.data = data
        self.next = None


def print_list(head: LinkedList) -> None:
    while head:
        print(head.data, end=" -> ")
        head = head.next
    print("None")


def merge_lists(headA, headB):
    dummy = LinkedList()
    curr = dummy

    while headA != None and headB != None:
        if headA.data < headB.data:
            curr.next = headA
            headA = headA.next
        else:
            curr.next = headB
            headB = headB.next

        curr = curr.next

    if headA != None:
        curr.next = headA
    else:
        curr.next = headB

    return dummy.next



node1 = LinkedList(1)
node1.next = LinkedList(2)
node1.next.next = LinkedList(7)


node2 = LinkedList(1)
node2.next = LinkedList(3)

print(merge_lists(node1, node2)) # 1-> 1-> 2-> 3-> 7-> None

Answer 1

由于它是一个虚拟节点，并且您永远不会使用该节点的数据属性，因此您可以将任何内容作为参数传递，例如 None:

dummy = LinkedList(None)

或者，您可以指定提供参数是可选的，并按如下方式定义构造函数：

class LinkedList:
    def __init__(self, data=None):
        self.data = data
        self.next = None

不相关，但在你的脚本末尾有：

print(merge_lists(node1, node2))

这将打印对象引用。您可能想调用为此目的定义的函数：

print_list(merge_lists(node1, node2))

如果您希望print 像那样工作，那么代替print_list 函数，使用__iter__ 方法丰富LinkedList 以简化对列表中值的迭代，以及__repr__ 或@ 987654331@方法如下：

class LinkedList:
    def __init__(self, data=None):
        self.data = data
        self.next = None

    def __iter__(self):
        head = self
        while head:
            yield head.data
            head = head.next
        yield None  # Optional

    def __repr__(self):
        return " -> ".join(map(str, self))

...然后你就可以了

print(merge_lists(node1, node2))