114.二叉树展开为链表

114.二叉树展开为链表

1.题目

2.我的解决方案

• 使用前序遍历,将其转化成链表,然后再将其构造成题目需要的二叉树
• 思路简单,代码不再赘述
• 时间复杂度:$O(n)$
• 空间复杂度:$O(n)$

3.官方的解决方案

• 更精妙的解决方案

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def flatten(self, root: TreeNode) -> None:
        """
        Do not return anything, modify root in-place instead.
        """
        """
        当前结点为cur
        从左子树开始
        先序遍历,若cur有左子树,需要进行展开,若没有左子树,对右子树进行展开cur = cur.right
        有左子树时, cur的左子树中最后一个遍历的就是cur左子树中最右边的结点
        遍历完最右边的这个结点,就开始遍历cur的右子树:故可以右子树挂到这个结点上,称此右节点为前驱(pre)
        然后将cur的整个左子树挂到cur的右子树
        cur的左子树置空
        对每个结点如此操作,可达成效果
        """
        cur = root
        while cur:
            if cur.left:    # 当左子树存在时
                pre = nxt = cur.left
                while pre.right:
                    pre = pre.right # 找到前驱
                pre.right = cur.right   # 将右子树给前驱
                cur.left = None # 左子树置空
                cur.right = nxt # 将左子树转变为右子树
            else:
                cur = cur.right # 从右子树开始处理

• 时间复杂度:$O(n)$
• 空间复杂度:$O(1)$