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

力扣

给定两个整数数组 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 <= preorder.length <= 3000
inorder.length == preorder.length
-3000 <= preorder[i], inorder[i] <= 3000
preorder 和 inorder 均 无重复 元素
inorder 均出现在 preorder
preorder 保证 为二叉树的前序遍历序列
inorder 保证 为二叉树的中序遍历序列

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/construct-binary-tree-from-preorder-and-inorder-traversal
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

前序后后序能够帮我们划分二叉树的根，而我们的中序遍历能够帮我们划分二叉树的左右子树。

思路：前序创建树，中序分隔左右子树 

 

preorder = [3,9,20,15,7]

inorder =    [9,3,15,20,7] 

从preoder中我们知道3就是跟，然后从inoder中，3左边的都是左子树，3右边的都是右子树。 

 

9的左和9和3之间没有数据了，也就是划分结束了 

然后我们看preoder中的9的右边，是20，也就是20是我们3的右子树的根，然后我们从inoder中知道20左边的15是我们20的左子树，20的右边的7是我们20的右子树 

所以我们就成功将这棵树构建出来了。

 

/**
 * 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:
    //这里的prev是通过前序去确定根。
    //inbegin和inend是我们的中序的容器的索引范围
    TreeNode* _buildTree(vector<int>& preorder, vector<int>& inorder,int& prei,int inbegin,int inend) {
        //如果已经无法创建了，就返回空
        //子树区间确认是否继续创建子树。
        if(inbegin>inend)
        {
            return nullptr;
        }
        //这道题还是一个前序遍历
        //每创建一个就将prei++
        TreeNode* root=new TreeNode(preorder[prei++]);
        //分隔中序
        int ini=inbegin;
        //这里我们给的是闭区间，所以是<=
        while(ini<=inend)
        {
            //我们已经找到了
            if(inorder[ini]==root->val)
            {
                //直接跳出
                break;
            }
            else
            {
                ++ini;
            }
        }
        //这样我们的区间就被划分成了这样三段
        //[inbegin,ini-1] ini [ini+1,inend]
        //创建左子树
        root->left=_buildTree(preorder,inorder,prei,inbegin,ini-1);
        //创建右子树
        root->right=_buildTree(preorder,inorder,prei,ini+1,inend);
        //最后将我们的root，也就是我们新创建的结点返回给上一层递归调用
        return root;
    }
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        int i=0;
        //这里我们调用上面的_buildTree函数，最初始的状态，也就是我们的分区是从0到我们的整一个容器的长度-1的范围
        return _buildTree(preorder,inorder,i,0,inorder.size()-1);
    }
};