AcWing 18. 重建二叉树（C++）- dfs

题目链接：https://www.acwing.com/problem/content/description/23/
题目如下：

思路：

写法一：y总

/**
 * 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:
    vector<int> pre,in;//为了方便，用作全局变量，因为递归中也要使用两个vector
    map<int,int> hash;//用来记录中序遍历中每个数的位置
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        pre=preorder,in=inorder;
        
        for(int i=0;i<in.size();i++)
            hash[in[i]]=i;
            
        return dfs(0,pre.size()-1,/**/0,in.size()-1);//前两个值为preorder中下标的位置，后两个值为inorder中下标的位置
    }
    
    TreeNode* dfs(int pre_left,int pre_right,/**/int in_left,int in_right){
        if(pre_left>pre_right) return NULL;//当前区间为0，返回nullptr
        
        TreeNode* root=new TreeNode(pre[pre_left]);//先序遍历的第一个元素为“树根”节点
        int pos=hash[root->val];//得到当前树根在中序遍历中的下标，且左半部分为左子树，右半部分为右子树
        
        root->left=dfs(pre_left+1,(pre_left+1)+(pos-in_left)-1,/**/in_left,pos-1);
        root->right=dfs((pre_left+1)+(pos-in_left)-1+1,pre_right,/**/pos+1,in_right);
        
        return root;
    }
};

写法二：卡哥

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        //前序(根左右)  中序(左根右)
        if(preorder.size()==0||inorder.size()==0) return nullptr;
        //6个参数分别为，前序-前开始-前末尾，中序-中开始-中末尾
        return dfs(preorder,0,preorder.size(),inorder,0,inorder.size());
    }
    TreeNode* dfs(vector<int>& pre,int preBegin,int preEnd,vector<int>& in,int inBegin,int inEnd){
        //1、先序
        if(preBegin==preEnd) return nullptr;//空节点

        int rootval=pre[preBegin];
        TreeNode* root=new TreeNode(rootval);

        if(preEnd-preBegin==1) return root;//只有一个节点

        //2、中序
        int index;
        for(index=inBegin;index<inEnd;index++){
            if(in[index]==rootval) break;//找到先序节点对应在中序中所在的位置
        }

        //3、分割
        //前序
        int leftpreBegin=preBegin+1;
        int leftpreEnd=preBegin+1+index-inBegin;
        int rightpreBegin=preBegin+1+(index-inBegin);
        int rightpreEnd=preEnd;
        //中序
        int leftinBegin=inBegin;
        int leftinEnd=index;
        int rightinBegin=index+1;
        int rightinEnd=inEnd;

        root->left=dfs(pre,leftpreBegin,leftpreEnd,in,leftinBegin,leftinEnd);
        root->right=dfs(pre,rightpreBegin,rightpreEnd,in,rightinBegin,rightinEnd);

        return root;
    }
};