LeetCode--114--二叉树展开为链表(python)

给定一个二叉树，原地将它展开为链表。

例如，给定二叉树

　　1
　　/ \
　  2   5
　 / \     \
　3 4      6
将其展开为：

　　　　1
　　　　 \
　　　  　2
　　　　   \
　　　　    3
　　　　　  \
　　　　       4
　　　　　　  \
　　　　　　　5
　　　　　　　  \
　　　　　　　　6

将root的右子树放到root的左子树的最右边作为右孩子         将root的左孩子变为自己的右孩子 (root.left=None)           root = root.right

　　1                                  1                                         1       
　　/ \                               / \                                            \ 
　  2   5                           2                                               2
　 / \     \                         / \                                               / \ 
　3 4      6                     3  4                                             3  4   
　　                                   \                                                    \   
　                                         5                                                 5
                                              \                                                   \   
　                                            6                                                  6

 class Solution:
     def flatten(self, root: TreeNode) -> None:
         """
         Do not return anything, modify root in-place instead.
         """
         if root==None or root.left == None and root.right == None :
             return root
         while(root != None):
             if(root.left == None):
                 root = root.right
             else:
                 pre = root.left
                 while pre and pre.right!=None:
                     pre = pre.right
                               pre.right = root.right
                 root.right = root.left
                 root.left = None
                 root = root.right