递归实现重建二叉树

Posted 2022-05-04 敲代码的Messi

输入某二叉树的前序遍历和中序遍历的结果，请构建该二叉树并返回其根节点。

假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

class Solution

public:
    TreeNode *buildTree(vector<int> &preorder, vector<int> &inorder)
    
        int len_pre = preorder.size();
        int len_in = inorder.size();
        if (len_pre == 0 || len_in == 0)
        
            return NULL;
        

        // 获得根节点
        TreeNode *root = new TreeNode(preorder.front());
        root->left = NULL;
        root->right = NULL;

        // 当前根结点为叶子节点
        if (len_pre == 1 && len_in == 1)
        
            if (preorder.front() == inorder.front())
            
                return root;
            
            throw invalid_argument("invalid input");
        

        // 在中序遍历中找到 根结点
        TreeNode *temp = root;
        auto index_root = find(inorder.begin(), inorder.end(), root->val);
        if (index_root == inorder.end()) //中序 序列中没有根节点
        
            throw invalid_argument("invalid input");
        

        if (index_root != inorder.begin()) //有左子树
        
            int len_left=index_root-inorder.begin(); //左子树长度
            vector<int> left_pre; //左子树先序遍历序列
            vector<int> left_in; //左子树中序遍历 序列
            left_in.resize(len_left);
            left_pre.resize(len_left);
            copy(preorder.begin()+1,preorder.begin()+1+len_left,left_pre.begin());
            copy(inorder.begin(),index_root,left_in.begin());

            root->left= buildTree(left_pre,left_in);
        
        if(index_root!=(inorder.end()-1))
            int len_right=inorder.end()-1-index_root; //右子树长度
            vector<int> right_pre;
            vector<int> right_in;
            right_in.resize(len_right);
            right_pre.resize(len_right);

            copy(index_root+1,inorder.end(),right_in.begin());
            copy(preorder.end()-len_right,preorder.end(),right_pre.begin());

            root->right=buildTree(right_pre,right_in);

        
        return root;
    
;