根据前中序列构造二叉树

力扣题目

图示分析

代码实现

C语言代码实现：

//构造树
struct TreeNode* buildTree(int* p, int pSize, int* in, int inSize){
    return CreatTree(p,in,0,0,pSize-1);
}
struct TreeNode*CreatTree(int*p,int*in,int index,int left,int right)
{
    //在left到right的范围内构造树,p为前序数组,in为中序数组,index为待构建根在前序p中下标
    if(left>right)
    	return NULL;
    
    //创建根节点
    struct TreeNode*root=(struct TreeNode*)malloc(sizeof(struct TreeNode));
    root->val=p[index];
    root->left=NULL;
    root->right=NULL;
    
    //找到根节点在中序in中的位置pos,找pos的目的一是为了找到右孩子，而是为了划分构建返回
    int pos=left;
    while(left<right&&p[index]!=in[pos])
    {
        pos++;
    }
    
    root->left=CreatTree(p,in,index+1,left,pos-1);//连接左孩子，并构造左侧区域子树
    root->right=CreatTree(p,in,pos+1-left+index,pos+1,right);//连接右孩子，并构造右侧区域子树
    
    return root;
}