leetcode114

class Solution {
public:
    void flatten(TreeNode* root) {
        while(root){
            if(root->left){
                TreeNode* pre=root->left;
                while(pre->right){
                    pre=pre->right;
 
                }
                pre->right=root->right;
                root->right=root->left;
                root->left=nullptr;
            }
            root=root->right;
        }
    }
};

补充一个DFS遍历，然后倒叙组装成类似链表的解决方案，使用python实现：

class Solution:
    def preOrder(self,root,l):
        if root != None:
            l.append(root)
 
            self.preOrder(root.left,l)
            self.preOrder(root.right,l)
 
    def flatten(self, root: 'TreeNode') -> 'None':
        if root == None:
            return
        l = list()
        self.preOrder(root,l)
        root = l[len(l)-1]
        for i in range(len(l)-2,-1,-1):
            l[i].left = None
            l[i].right = root
            root = l[i]

python的递归实现：

 class Solution:
     def build(self,root):
         if root != None:
             if root.left != None and root.right != None:
                 right = self.build(root.right)#重建右子树
                 left = self.build(root.left)#重建左子树
                 leaf = left
                 while leaf.right != None:#找到左子树的叶子节点
                     leaf = leaf.right
                 leaf.right = right#右子树连接到左子树的末尾
                 root.right = left#根节点修改右子树
                 root.left = None#根结点左子树设为空
                 return root
             elif root.left != None and root.right == None:
                 root.right = self.build(root.left)
                 root.left = None
                 return root
             elif root.left == None and root.right != None:
                 root.right = self.build(root.right)
                 return root
             else:
                 return root
         return None
 
     def flatten(self, root: TreeNode) -> None:
         """
         Do not return anything, modify root in-place instead.
         """
         self.build(root)