【剑指offer】面试题7:重建二叉树【C++版本】

总结的部分题目思路与代码，待完善。
【剑指offer-第二版】部分题目与解答【C++版本】

题目：

重建二叉树

输入某二叉树的前序遍历和中序遍历的结果，请重建该二叉树（假设没有重复数字）。树节点定义如下：

struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) :val(x), left(nullptr), right(nullptr) {}
};

解题思路：

1.前序遍历的第一个数字总是树的根节点的值。但在中序遍历中，根节点的值在序列的中间，其中左子树的节点的值位于根节点的值的左边，右子树的节点的值位于根节点的值的右边。要扫描中序遍历序列，才能找到根节点的值。
2.又因为前序遍历总是在根节点后先遍历完左子树，才会遍历右子树，所以由后续遍历推断出左子树中节点的数目之后，前序遍历中根节点之后的相同数目的值都是左子树的节点的值。剩下的就是右子树的值。这样就分别找到了左右子树序列。
3.分别确定了根节点和左右子树的前序、中序遍历，我们可以用同样的方法构建左右子树，剩下的可以用递归来完成。

可以AC的解法【C++版本】

#include <vector>
#include <iostream>
#include <string>
#include <stdlib.h>
#include <algorithm>

using namespace std;

struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) :val(x), left(nullptr), right(nullptr) {}
};

TreeNode* reConstructBinaryTree(vector<int> pre, vector<int> vin);

int main()
{
    vector<int> pre{1,2,4,5,3,6,7};
    vector<int> vin{4,2,5,1,6,3,7};

    TreeNode *newRoot = reConstructBinaryTree(pre, vin);

    system("pause");
    return 0;
}

TreeNode* reConstructBinaryTree(vector<int> pre, vector<int> vin)
{
    if (pre.size() == 0 || pre.size() != vin.size())
        return nullptr;

    int root = pre[0];
    TreeNode *newNode = new TreeNode(root);
    //如果只剩一个节点了，那么可以直接返回
    if (pre.size() == 1)
        return newNode;

    auto posi = find(vin.begin(), vin.end(), root);
    //错误检测
    if (posi == vin.end()) {
        return nullptr;
    }
    int leftSize = posi - vin.begin();
    int rightSize = vin.end() - posi - 1;

    //递归求解
    //这里取前序和后序遍历的左右子树可能有点绕，可以好好思考一下
    newNode->left = reConstructBinaryTree(vector<int>(pre.begin() + 1, pre.begin() + 1 + leftSize), 
        vector<int>(vin.begin(), vin.begin() + leftSize));
    newNode->right = reConstructBinaryTree(vector<int>(pre.begin() + 1 + leftSize, pre.end()), 
        vector<int>(vin.begin() + leftSize + 1, vin.end()));

    return newNode;
}