Leetcode105.从前序遍历与中序遍历构造二叉树

题目描述

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

示例

示例 1:

输入: 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]

思路分析

1: 用前序数组去创建根,然后遍历中序数组,找到与根创建相同的结点,并记录这个结点的下标.

2: 通过这个下标就可以找出左右区间,左子树的根一定在左区间,右子树的根一定在右区间.

3:分别递归创建左右子树,并于根的左右结点相连接.

4:当左右子树创建完毕后,此时便会回溯到第一个函数栈帧中右子树递归完毕的位置,此时我们要返回根结点.

5: 这道题递归时返回上一层的条件即创建根的条件区间一定要合规.

代码实现

class Solution {
public:

    TreeNode* _buildTree(vector<int>& preorder,vector<int>& inorder,int& prei,int inbegin,int intend)
    {
        //如果区间不符合条件，就不能再创建结点了。
        if(  inbegin > intend )
        {
            return nullptr;
        }
        //用前序的第一个值去创建根。
        TreeNode* root = new TreeNode(preorder[prei++]);
        //分割中序。
        int ini = inbegin;
        while( ini <= intend )
        {
            //如果找到了
            if( inorder[ini] == root->val )
            {
                break;
            }
            //否则就继续inorder与order中相同的根结点，进而分出根的左右区间。
            else
            {
                ++ini;
            }
        }
        //走到这里，就说明已经找到了,此时的区间为[inbegin,ini -1] ini [inti+1,intend],如果还有区间，
        //就说明还能继续向左递归,继续生成新的结点，如果区间不合格，就说明这个结点的左右子树都为空。
        root->left = _buildTree( preorder,inorder,prei,inbegin,ini-1);
        root->right = _buildTree( preorder,inorder,prei,ini+1,intend);
        
        //左右遍历完，最后回溯到第一个root结点返回。
        return root;

    }
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder)
     {
        
        int i = 0;
        return _buildTree( preorder,inorder,i,0,inorder.size()-1);
    }
};