博客

leetcode


输入一棵二叉搜索树，将该二叉搜索树转换成一个排序的循环双向链表。要求不能创建任何新的节点，只能调整树中节点指针的指向。

思路：

得到一个排序的数组，二叉搜索树的特点是，左子树<=根节点<=右子树，用中序遍历二叉搜索树就可以得到一个递增数列了

对于循环双向列表，先构造双向列表，再将头尾相连

"""
# Definition for a Node.
class Node:
    def __init__(self, val, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right
"""
class Solution:
    def __init__(self):
        self.pre = None
        self.head = None
        self.tail = None

    def treeToDoublyList(self, root: 'Node') -> 'Node':
        if not root:
            return 
        def helper(root):
            if not root:
                return None
            #左子树，根节点，右子树，递增序列
            helper(root.left)
            #根节点
            if not self.pre:
                self.head = root
            else:
                self.pre.right = root
            root.left =self.pre
            self.pre = root
            self.tail = root



            helper(root.right)
        helper(root)
        #头尾相连
        self.head.left = self.tail
        self.tail.right = self.head
        return self.head