牛客网 剑指offer 4 重建二叉树

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/abc15766228491/article/details/82286251

题目描述
输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建二叉树并返回。

首先将前序的第一个数字拿出来，这个是二叉树的根，在中序中找到这个数字，将中序分为左右两部分，即为左子树和右子树。再将统计中序分出的左右子树的长度，从先序中除去第一个开始，把后面的页分为两部分，即为新的左右子树的先序遍历，然后用左子树的先序和左子树的中序推左子树，右子树的先序和右子树的中序推右子树，递归即可

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* reConstructCore(vector<int> pre, vector<int> vin){
    int root_value = pre[0];
    TreeNode* root = new TreeNode(root_value);
    vector<int>::iterator iter1 = find(vin.begin(), vin.end(), root_value);
    int leftTree_length = iter1-vin.begin();
    vector<int> new_leftPre, new_rightPre;
    for (int i = 1; i < pre.size(); ++i) {
        if(i<=leftTree_length)
            new_leftPre.push_back(pre[i]);
        else
            new_rightPre.push_back(pre[i]);
    }
    vector<int> new_leftVin, new_rightVin;
    int flag = 0;
    for (int i = 0; i < vin.size(); ++i) {
        if(i==leftTree_length) {
            //这里不是leftTree_length-1
            flag=1;
            continue;
        }
        if(flag==0) new_leftVin.push_back(vin[i]);
        else new_rightVin.push_back(vin[i]);
    }
    if(new_leftPre.size()>0)
    root->left = reConstructCore(new_leftPre, new_leftVin);
    if(new_rightPre.size()>0)
    root->right = reConstructCore(new_rightPre, new_rightVin);
    return root;
}
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
        if(pre.size()==0 || vin.size()==0)
        return nullptr;
    TreeNode* root = reConstructCore(pre, vin);
    return root;
    }
};