Leetcode 426. 将二叉搜索树转换为有序双向链表？

Question

对于leetcode上的426号问题我很困惑，我觉得我的回答是对的。但运行后显示我错了。以下是问题和我的原始答案：

"""
# Definition for a Node.
class Node:
    def __init__(self, val, left, right):
        self.val = val
        self.left = left
        self.right = right
"""
class Solution:
    def treeToDoublyList(self, root):
        """
        :type root: Node
        :rtype: Node
        """
        if root:
            sign = True
            stack = []
            node = root

            while stack or node:
                if node:
                    stack.append(node)
                    node = node.left 
                else:
                    node = stack.pop()

                    if sign:
                        pre,head = node, node
                    else:
                        pre.right = node
                        node.left = pre
                        pre = node

                    node = node.right

            head.left = pre
            pre.right = pre

            return head
        else:
            return None

有人可以帮我弄清楚我的代码有什么问题吗？任何意见或建议将不胜感激。

Answer 1

我发现代码有两个问题。

首先是在您的if sign: 块内，您需要设置sign = False，因为您只想初始化head 一次并且只在第一次执行该块。 （不知道为什么将变量称为sign，也许first 会更合适，或者只是在这种情况下重用head = None 也可以。）

第二个错误更小，影响列表中的最后一个链接，使其循环。您想设置pre.right = head 而不是pre，以便列表的最后一个节点指向第一个节点，而不是自身。

我还没有真正测试过这个，所以我可能遗漏了一些东西，但在我看来，这应该足以修复此代码。