【hot100】刷题记录(20)-二叉树展开为链表

题目描述：

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

 

 

我的作答：

从右往左移动，如示例1，先把5,6移动到4下面，再把4移动到3下面，再把2移动到右边；

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution(object):
    def flatten(self, root):
        """
        :type root: Optional[TreeNode]
        :rtype: None Do not return anything, modify root in-place instead.
        """
        cur = root
        while cur:
            if cur.left:
                p = cur.left #指向根节点的左孩子
                while p.right: p = p.right #一直遍历到该子树的最右孩子
                p.right = cur.right #把根节点的右孩子挂到下面来
                cur.right = cur.left #把根节点的左孩子变成右孩子
                cur.left = None
            cur = cur.right #移动到子树

 

参考：

无