将二叉树转换为双向链表。如何避免在这里使用全局变量？

Question

转换函数将二叉树转换为链表。由于使用全局变量，我无法通过 geeksforgeek 中的测试用例。如何避免使用全局变量？ prev = None head = Nonedef convert(root):    global prev    global head    if root is None:        return     convert(root.left)    if head== None:        head = root        prev = root    else:        root.left = prev        prev.right = root    prev = root    convert(root.right)

Answer 1

将它们作为参数传递，默认 = None：

def convert(root, prev = None, head = None):

    if root is None:

        return 

    convert(root.left, prev=prev, head=head)

    if head== None:

        head = root

        prev = root

    else:

        root.left = prev

        prev.right = root

    prev = root

    convert(root.right, prev=prev, head=head)

Answer 2

我认为你需要返回链表的头。

不管怎样，你可以包装你的函数来为这两个变量创建一个闭包：

def convert(root):

    head = None

    prev = None

    def recur(cur):

        nonlocal prev, head

        if cur is None:

            return 

    

        recur(cur.left)

        if head is None:

            head = cur

        else:

            cur.left = prev

            prev.right = cur

        prev = cur

        recur(cur.right)

    recur(root)

    return head