剑指offer07.重建二叉树

0517刷题自用

题目描述

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

代码

class Solution {
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        if(preorder.size()==0&&inorder.size()==0) return NULL;
        auto headroot=find(inorder.begin(),inorder.end(),preorder[0]);
        TreeNode*root=new TreeNode(preorder[0]);
        vector<int>inorderleft(inorder.begin(),headroot);
        vector<int>inorderright(headroot+1,inorder.end());
        int n=inorderleft.size();
        vector<int>preorderleft(preorder.begin()+1,preorder.begin()+1+n);
        vector<int>preorderright(preorder.begin()+1+n,preorder.end());
        root->left=buildTree(preorderleft,inorderleft);
        root->right=buildTree(preorderright,inorderright);
        return root;
    }
};

思路

先找到根节点，即前序遍历数组的第一个数
在中序遍历中找到根节点，根节点之前为左子树，之后为右子树
根据找出的左右子树找出前序遍历中的左右子树
左右子树递归
…