二叉树展开为链表

题目

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

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

示例 1：

输入：root = [1,2,5,3,4,null,6]
输出：[1,null,2,null,3,null,4,null,5,null,6]

示例 2：
输入：root = []
输出：[]

示例 3：
输入：root = [0]
输出：[0]

提示：

• 树中结点数在范围 [0, 2000] 内
• -100 <= Node.val <= 100

进阶：你可以使用原地算法（O(1) 额外空间）展开这棵树吗？

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/flatten-binary-tree-to-linked-list
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

后序遍历&递归

从递归出发，首先求解子问题，即把一个只有三个节点的树进行转换：

1. 右子节点拼到左子节点的右子节点
   root.left.right=root.right
2. 拼好的左子节点放到root的右子节点位置
   root.right=root.left
3. 左子节点设置为None
   root.left=None

当 root 的左节点和右节点都拼好，这时把root的右子节点root.right拼到root.left的末尾，用一个index指针，通过while循环去找root.left的末尾再进行拼接。

最后，递归完成全部操作。

方法二

直接递归，按照题目的要求进行判断和交换。

提交代码

后序遍历&递归

# 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)
        if root.left and root.right:
            # 寻找root.left尾部用的指针
            index = root.left
            # while循环找root.left的尾部
            while index.right:
                index=index.right
            # 进行拼接工作
            index.right = root.right
            root.right=root.left
            root.left=None
            # 如果没有左节点，就不用拼接了
        elif not root.right and root.left:
            root.right=root.left
            root.left=None

结果

方法二

# 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.
        """
        while root:
            if root.left:   #左子树存在的话才进行操作
                sub_left = root.left
                while sub_left.right:   #左子树的右子树找到最深
                    sub_left = sub_left.right
                sub_left.right = root.right #将root的右子树挂到左子树的右子树的最深
                root.right = root.left      #将root的左子树挂到右子树
                root.left = None            #将root左子树清空
            root = root.right               #继续下一个节点的操作

结果

学习

1. 从示例得到题目意图，对二叉树的展开顺序和规则进行分析；
2. 方法一的思路将解题顺序拆解，while循环写在条件判断里，方法二代码简洁，while循环写在外层，原理相同，均需要深入到 root 左节点的右节点末尾，重复展开过程。