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
             16                 pre.right = root.right
                root.right = root.left
                root.left = None
                root = root.right