LeetCode 114. Flatten Binary Tree to Linked List

Posted 2021-01-01 約束の空

树里的 divide and conquer 感觉和 post order traversal 就是一个东西，反正都是递归。

下面方法返回了 root ，也可以不返回，直接用 flatten 自己递归也行。

时间复杂度 O(nlogn) 空间复杂度 O(h)

class Solution {
public:
    
    void flatten(TreeNode* root) {  
        helper(root);
    }
    
    TreeNode *helper(TreeNode* root) {  
        if (root==NULL) return NULL;
        TreeNode *left=helper(root->left);
        TreeNode *right=helper(root->right);
        root->left = NULL;
        root->right = left;
        TreeNode *cur=root;
        while (cur->right!=NULL) cur=cur->right;
        cur->right = right;
        return root;
    }

};