【LeetCode】-- 105. 从前序与中序遍历序列构造二叉树

1. 题目

给定两个整数数组 preorder 和 inorder ，其中 preorder 是二叉树的先序遍历， inorder 是同一棵树的中序遍历，请构造二叉树并返回其根节点。

2. 示例

示例一：

输入: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
输出: [3,9,20,null,null,15,7]

示例 2:

输入: preorder = [-1], inorder = [-1]
输出: [-1]

3. 分析

1.前序和中序可以构造二叉树，中序和后续也可以构造二叉树，但是前序和后序不能构造二叉树。

2.因为有中序就可以确定左子树和右子树区间。而用前序下标可以确定节点在根的左子树还是右子树。所以中序用区间，前序用下标

3.用前序第一个节点做根，再去中序里面找这个根节点，中序里这个节点的左区间就是它的左子树，右区间就是它的右子树，递归左子树与右子树，就可以构建这棵树。递归过程如下：

 

4. 代码实现

/**
 * 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,int& pi,vector<int>& inorder,int inBegin,int inEnd)
    {
        if(inBegin > inEnd)
        {
            return nullptr;
        }

        TreeNode* root = new TreeNode(preorder[pi]);
        ++pi;
        int rooti = inBegin;

        //在中序数组中找前序第pi个节点
        while(rooti <= inEnd)
        {
            if(root->val == inorder[rooti])
            {
                break;
            }
            else
            {
                ++rooti;
            }
        }

        //找到后就递归中序的左子树区间和右子树驱区间
        root->left = _buildTree(preorder,pi,inorder,inBegin,rooti-1);
        root->right = _buildTree(preorder,pi,inorder,rooti+1,inEnd);
        return root;
    }
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) 
    {
        int i = 0;//递归的入参，递归过程中需要修改
        return _buildTree(preorder,i,inorder,0,inorder.size()-1);
    }
};