剑指offer-重建二叉树

问题

题目：[重建二叉树]

思路

老问题了，递归建立。
基本思路是，根据先序中的第一个节点。到中序序列去进行划分左右子树。
然后递归建立左右子树即可。

代码

/** * 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* reConstructBinaryTree(vector<int> pre,vector<int> vin) {        return create( pre, 0, pre.size()-1, vin, 0, vin.size()-1 );           }private:    TreeNode* create(const vector<int>& pre, int s1, int t1, const vector<int>& in, int s2, int t2) {        if(s1 > t1) return NULL;        TreeNode* root = new TreeNode(pre[s1]);        int mid = 0;        for(mid = s2; mid <= t2; ++mid){            if(in[mid] == pre[s1]) break;        }        int left_len = mid-s2;        root->left = create( pre, s1+1, s1 + left_len, in, s2, mid-1 );        root->right = create(pre, s1+left_len+1, t1, in, mid+1, t2);        return root;    }};