leetcode114.二叉树展开为链表

leetcode 114.二叉树展开为链表

给你二叉树的根结点 root ，请你将它展开为一个单链表：

• 展开后的单链表应该同样使用 TreeNode ，其中 right 子指针指向链表中下一个结点，而左子指针始终为 null 。
• 展开后的单链表应该与二叉树 先序遍历 顺序相同。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/flatten-binary-tree-to-linked-list

解题思路

题目解读：就是将左子树和右子树展开为链表之后，先将左子树放到根节点的右边再将右子树放入原左子树的右边

这样的话就是用递归来处理

# 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.
        """
        if not root:
            return

        self.flatten(root.left)
        self.flatten(root.right)

        # 将root右子树暂存
        temp = root.right
        # 左子树放到右边
        root.right = root.left
        # 将左子树置空
        root.left = None
        # 找到原左子树的最右边
        while root.right:
            root = root.right
        root.right = temp