重建二叉树

重建二叉树

题目描述

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

没做出来, 抄牛客讨论里的答案, 解题关键是i-startIn这个在中序中的偏移量, 这个偏移量在前序中所偏移包括的元素正好是中序中的左子树

/**
 * 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) {
        TreeNode *ret = constructTree(pre, 0, pre.size() - 1, vin, 0, vin. size() - 1);
        
        return ret;
    }
private: 
    TreeNode* constructTree(vector<int> pre, int startPre, int endPre, vector<int> in, int startIn, int endIn) {
        if ((startPre > endPre) || (startIn > endIn)) {
            return NULL;
        }
        
        TreeNode* ret = new TreeNode(pre[startPre]);
        
        for (int i = 0; i <= endIn; i++) {
            if (pre[startPre] == in[i]) {
                ret->left = constructTree(pre, startPre+1, startPre+(i-startIn), in, startIn, i-1);
                ret->right = constructTree(pre, startPre+(i-startIn)+1, endPre, in, i+1, endIn);
                break;
            }
        }
        
        return ret;
    }
    
};