重建二叉树

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

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

class Solution {
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        TreeNode* root;
        if (preorder.empty())
        {
            return nullptr;
        }
        else
        {
            root = new TreeNode;
            root->val = preorder[0];
            auto it = find(inorder.begin(), inorder.end(), preorder[0]);
            int n = it - inorder.begin();//前序长度
            vector<int>pre1;//当前根的左边的前序数组
            pre1.assign(preorder.begin() + 1, preorder.begin() + n+1);
            vector<int>in1;//当前根的左边的中序数组
            in1.assign(inorder.begin(), it);
            vector<int>pre2;//当前根的右边的前序数组
            pre2.assign(preorder.begin() + n+1, preorder.end());
            vector<int>in2;//当前根的右边的中序数组
            in2.assign(it + 1, inorder.end());
            root->left = buildTree(pre1, in1);
            root->right = buildTree(pre2, in2);
        }
        return root;
    }
};```

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值