给你二叉树的根结点 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]
提示:
进阶:你可以使用原地算法(O(1) 额外空间)展开这棵树吗?
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/flatten-binary-tree-to-linked-list
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
从递归出发,首先求解子问题,即把一个只有三个节点的树进行转换:
当 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 #继续下一个节点的操作