老卫带你学---leetcode刷题(114. 二叉树展开为链表)

114. 二叉树展开为链表

问题:

给你二叉树的根结点 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

你可能感兴趣的:(leetcode,链表)