剑指 Offer 07. 重建二叉树

输入某二叉树的前序遍历和中序遍历的结果，请构建该二叉树并返回其根节点。

假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

struct TreeNode* buildTree(int* preorder, int preorderSize, int* inorder, int inorderSize)
{
    if(preorderSize ==0 || inorderSize==0 || preorderSize!=inorderSize)
        return NULL;
    int leftNum=0;
    int rightNum=0;
    int rootVal=preorder[0];//根结点

    //计算左结点个数
    for(int i=0;i<inorderSize;i++)
    {
        if(inorder[i]==rootVal)
        {
            break;
        }
        leftNum++;
    }
    //计算右节点个数
    rightNum=preorderSize-leftNum-1;

    //构建二叉树
    struct TreeNode* root=malloc(sizeof(struct TreeNode));
    root->val=rootVal;
    root->left=buildTree(&preorder[1],leftNum,&inorder[0],leftNum);
    root->right=buildTree(&preorder[1+leftNum],rightNum,&inorder[1+leftNum],rightNum);
    return root;

}

Input: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
Output: [3,9,20,null,null,15,7]

示例 2:

Input: preorder = [-1], inorder = [-1]
Output: [-1]