0
点赞
收藏
分享

微信扫一扫

递归实现重建二叉树

在觉 2022-05-02 阅读 58

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

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

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;
    }
};
举报

相关推荐

0 条评论