剑指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) {
        int len=vin.size();
        if(len==0)return nullptr;
        
        //创建根节点，根节点肯定是前序遍历的第一个数
        TreeNode *root=new TreeNode(pre[0]);
        int loc=0;//找到中序遍历根节点所在位置,存放于变量loc中
        while(vin[loc]!=pre[0]){
            loc++;
        }
        
        //对于中序遍历，根节点左边的节点位于二叉树的左边，根节点右边的节点位于二叉树的右边
        //利用上述这点，对二叉树节点进行归并
        vector<int> leftPre,leftVin,rightPre,rightVin;
        for(int i=0;i<loc;i++){
            leftPre.push_back(pre[i+1]);
            leftVin.push_back(vin[i]);
        }
        
        for(int j=loc+1;j<len;j++){
            rightPre.push_back(pre[j]);
            rightVin.push_back(vin[j]);       
        }
        
        //和shell排序的思想类似，取出前序和中序遍历根节点左边和右边的子树
        //递归，再对其进行上述所有步骤，即再区分子树的左、右子子树，直到叶节点
        root->left=reConstructBinaryTree(leftPre,leftVin);
        root->right=reConstructBinaryTree(rightPre,rightVin);
        
        return root;
    }
};