105. 从前序与中序遍历序列构造二叉树(深搜)

注意: 二叉树中没有重复元素,有重复元素就搞不出来了.

/**
 * 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* root;
    vector<int> Pre, In; //搞两个全局变量,这样调用函数的时候就不用传参了
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        //pre:前序 in:中序
        if (preorder.empty() || inorder.empty()) return 0;
        Pre = preorder;
        In = inorder;
        build(0, Pre.size() - 1, 0, In.size() - 1, root);    
        return root;
    }
    //build函数这个TreeNode*&引用很关键,不加这个&,函数栈里的root就只会传值,不会对原root进行修改
    void build(int P1,int P2,int I1,int I2,TreeNode*& root) {
        if (P1 > P2 || I1 > I2) return;
        int P = 0, len = 0; //P存储数左右子树的父亲节点,len存储左子树的节点数
        root = new TreeNode(Pre[P1]); //左右子树父亲节点
        for (int i = I1; i <= I2; i++) {
            if (In[i] == Pre[P1]) {
                P = i;  //找到左右子树的分割索引
                break;
            }
        }
        if (In[P] != Pre[P1]) return;
        len = P - I1;  //左子树的节点数
        build(P1 + 1, P1 + len, I1, P - 1, root->left); //建立左子树
        build(P1 + len + 1, P2, P + 1, I2, root->right);//建立右子树
    }
};