【剑指offer】036.二叉搜索树与双向链表

题目

【剑指offer】036.二叉搜索树与双向链表

解法

二叉搜索树相关的题目，首先联想到中序遍历，因为中序遍历的结果刚好是有序的，大部分题目是在中序遍历的基础上稍微调整。
这里用self.head和self.tail来记录。
self.head在递归的时候只要赋值一次就可以了。
self.tail会不停地变，当把root的左子树遍历完后，当前self.tail是指向左子树里面的最后一个节点。此时要把这个节点和root连接，并且把self.tail指向root。注意，此时不需要将self.tail和self.head 相连，只要在递归都结束后，将self.head和self.tail相连就可以了。

"""
# 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 treeToDoublyList(self, root: 'Node') -> 'Node':
        def dfs(cur):
            if not cur: return
            dfs(cur.left)
            if not self.head:
                self.head = cur
                self.tail = cur
            else:
                self.tail.right = cur
                cur.left = self.tail
                self.tail = cur
            dfs(cur.right)
        if not root: return None
        self.head, self.tail = None, None
        dfs(root)
        self.head.left = self.tail
        self.tail.right = self.head
        return self.head