剑指offer07_重建二叉树

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

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

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

Solution：

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

using namespace std;

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

class Solution{
public:
    TreeNode* reConstructBinaryTree(vector<int> pre, vector<int> vin){
        if ( pre.size() == NULL || pre.size() != vin.size())
            return NULL;

        int root = pre[0];
        TreeNode *newNode = new TreeNode(root);
        if(pre.size() == 1)
            return newNode;

        int vinRootPost = 0;
        for(int i = 0; i < vin.size(); i++){
            if (vin[i] == root )
                vinRootPost = i;
        }

        vector<int> a;
        for(int i = 1; i < vinRootPost+1;i++){
            a.push_back(pre[i]);
        }
        vector<int> b;
        for (int j = vinRootPost+1; j < pre.size(); j++) {
            b.push_back(pre[j]);
        }
        vector<int> c;
        for(int k = 0; k < vinRootPost;k++){
            c.push_back(vin[k]);
        }
        vector<int> d;
        for (int z = vinRootPost+1; z < vin.size(); z++) {
            d.push_back(vin[z]);
        }

        newNode->right = reConstructBinaryTree(b,d);
        newNode->left = reConstructBinaryTree(a,c);

        return newNode;

    }

    void postOrder(TreeNode *root){
        if(root->left)
            postOrder(root->left);
        if(root->right)
            postOrder(root->right);
        printf("%d", root->val);
    }

};

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

    TreeNode *newRoot = Solution().reConstructBinaryTree(pre, vin);
    Solution().postOrder(newRoot);

    return 0;
}