LeetCode 1028. 从先序遍历还原二叉树(递归建树、切割序列)

从先序遍历还原二叉树
做此题的思路和用 中序遍历和前(后)序遍历重建二叉树的思路是非常一致的。
从序列中找出根节点、左子树、右子树——不断切割序列,递归地向下建树
——而确定根节点和左右子树根节点的依据就是深度
时间复杂度:O(S.length)O(S.length)
但是不知道为什么这么慢啊

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* recoverFromPreorder(string S) {
        return create(0,0,S.size()-1,S);
    }
    TreeNode* create(int layer,int ld,int rd,const string &S){
        if(ld>rd){
            return nullptr;
        }
        
        int val = 0;
        for(int i = layer+ld;i<=rd;i++){
            if(S[i]<'0' || S[i]>'9'){
                break;
            }
            val = val*10+(S[i]-'0');
        }

        TreeNode* root = new TreeNode(val);
        string target = "";
        for(int i=0;i<layer+1;i++){
            target+="-";
        }
        int lidx =-1,ridx=-1;
        root->left = nullptr;
        root->right = nullptr;
        for(int i=ld;i+layer<=rd;i++){

            if(S.substr(i,layer+1) == target && ( i+layer+1>rd || S[i+layer+1]!='-') && (i-1<ld || S[i-1]!='-'  ) ){

                if(lidx==-1){
                    lidx = i;
                }else{
                    ridx = i;
                }
            }
        }   

        if(lidx==-1){
            return root;
        }
        if(ridx==-1){
            root->left = create(layer+1,lidx,rd,S); 
        }else{
            root->left = create(layer+1,lidx,ridx-1,S);
            root->right = create(layer+1,ridx,rd,S);
        }
        return root;
    }
};
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章