给你二叉树的根结点 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
class Solution:
def flatten(self, root: Optional[TreeNode]) -> None:
"""
Do not return anything, modify root in-place instead.
"""
while root:
if root.left: ##只有root有左子树再操作
sub_left = root.left
while sub_left.right: ## 寻找左子树的右子树的最深
sub_left=sub_left.right
## 比如[1,2,5,3,4,null,6]就是将5,6挂到4后面
sub_left.right=root.right ## 将root的右子树挂到左子树的右子树的最深
root.right=root.left ## 调整结构
root.left=None
root=root.right