DFS算法总结之从前序、中序确定二叉树

题目：
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
解题思路：
前序遍历第一遍历的点总是后面的根节点，而中序遍历根节点是在“中间”的位置.如例子，整棵树根节点肯定是3，而在中序遍历中，3的左边肯定左子树，右边是右子树。这样中序遍历分成分成【9】【根节点3】【15,20,7】三个部分，前序遍历也分成【根节点3】【9】【20,15,7】三个部分，将3作为根节点，【9】和【9】、【15,20,7】和【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:
    unordered_map<int,int> map;
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        int preR=preorder.size()-1;
        int inR=inorder.size()-1;
        for(int i=0;i<=inR;i++){
            map[inorder[i]]=i;//建立中序遍历的hash表
        }
        TreeNode* res=dfs(preorder,0,preR,inorder,0,inR);
        return res;
    }
    TreeNode* dfs(vector<int>&pre,int preL,int preR,vector<int>&ino,int inL,int inR){
        if(preL<=preR&&inL<=inR){//保证此时节点数至少是1
            int rootValue=pre[preL];
            int index=map[rootValue];//该节点在中序遍历的序号
            TreeNode* root=new TreeNode(rootValue);
            root->left=dfs(pre,preL+1,preL+index-inL,ino,inL,index-1);//左子树
            root->right=dfs(pre,preL+index-inL+1,preR,ino,index+1,inR);//右子树
            return root;    
        }
        else
            return NULL;
    }
};