LeetCode#105从前序中序构建二叉树

题目描述

根据一棵树的前序遍历与中序遍历构造二叉树。

注意:
你可以假设树中没有重复的元素。

例如，给出

前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树：

    3
   / \
  9  20
    /  \
   15   7

简单分析

学数据结构时候做过类似的，当时是字符串构建，然后想着用同样方法试试，然后一遍过√，但是空间利用太多了，先把暴力解放在这里，想想怎么优化。
补：可以不用申请空间，直接重写函数把长度作为参数传进去，然后就直接以该位置访问即可

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
    if(preorder.size()==0||inorder.size()==0){
        return NULL;
    }
    int index = preorder[0];
    TreeNode *root = new TreeNode(index);
    vector<int> left;
    for(int i=0;i<inorder.size();i++){
        if(inorder[i]==index){
            index = i;
            break;
        }
        else{
            left.push_back(inorder[i]);
        }
    }
    vector<int> preLeft;
    for(int i=1;i<=left.size();i++){
        preLeft.push_back(preorder[i]);
    }
    vector<int> right;
    for(int i = index+1;i<inorder.size();i++){
        right.push_back(inorder[i]);
    }
    vector<int> preRight;
    for(int i=1;i<=right.size();i++){
        preRight.push_back(preorder[i+left.size()]);
    }
    root->left = buildTree(preLeft,left);
    root->right=buildTree(preRight,right);

    return root;
}
};
//优化版
/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        int index = 0;
        return rebuildTree(preorder,inorder,0,preorder.size()-1,index);
    }

    TreeNode* rebuildTree(vector<int>& preorder, vector<int>& inorder,int left,int right,int & index) {
        if(index>=preorder.size()){
            return NULL;
        }

        TreeNode *root = new TreeNode(preorder[index]);

        int i=left;
        for(;i<=right;i++){
            if(inorder[i]==preorder[index]){
                break;
            }
        }

        if(left<=i-1){
            root->left = rebuildTree(preorder,inorder,left,i-1,++index);
        }

        if(i+1<=right){
            root->right= rebuildTree(preorder,inorder,i+1,right,++index);
        }

    return root;
    }
};

解题补注

补注：index的值直接作为数字传入则随着函数调用不会依次增加，而对某个root左右子树都一致，因为是深度构建树，需要**int &**进行变量引用，才能做到每一次函数调用，都达到对已处理节点计数的作用

引用