LeetCode--105 从前序与中序遍历序列构造二叉树 （Construct Binary Tree from Preorder and Inorder Traversal）（C语言版）

题目描述：

解题思路：

• 通过先序遍历来确定每次所要创建的根节点
• 通过 i 来将中序遍历的区间不断划分为左右区间
• 再分别在左右区间中继续执行上述步骤

代码如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */
struct TreeNode* buildTree(int* preorder, int preorderSize, int* inorder, int inorderSize) {
    //通过前序寻找根节点，通过中序寻找左右节点
    //判断是否需要创建节点，当inorderSize为0时，证明已经不需要在创建新的节点
    if(inorderSize==0)
        return NULL;
    //创建新节点
    struct TreeNode* root=(struct TreeNode*)malloc(sizeof(struct TreeNode));
    root->val=preorder[0];
    int i=0;
    //在中序遍历中寻找当前根节点
    for(;i<inorderSize;i++){
        if(inorder[i]==preorder[0])
            break;
    }
    //如果在中序遍历中没有找到当前根节点，则直接返回空
    if(i==inorderSize)
        return NULL;
    //创建左节点
    root->left=buildTree(preorder+1,preorderSize,inorder,i);
    //创建右节点
    root->right=buildTree(preorder+i+1,preorderSize,inorder+i+1,inorderSize-i-1);
    return root;
}

 

通过严格划分区间的方法

代码如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */
struct TreeNode* BuildTree(int* preorder,int pstart,int pend,int* inorder,int istart,int iend)
{
    //判断是否需要创建节点
    if(pstart>pend||istart>iend)
        return NULL;
    struct TreeNode* root=(struct TreeNode*)malloc(sizeof(struct TreeNode));
    root->val=preorder[pstart];
    int i=istart;
    //在中序遍历中查找当前根节点位置
    for(;i<=iend;i++){
        if(inorder[i]==preorder[pstart])
            break;
    }
    //若没有找到直接返回空
    if(i==iend+1)
        return NULL;
    //创建左节点
    root->left=BuildTree(preorder,pstart+1,pstart+i-istart,inorder,istart,i-1);
    //创建右节点
    root->right=BuildTree(preorder,pstart+i+1-istart,pend,inorder,i+1,iend);
    return root;
}
struct TreeNode* buildTree(int* preorder, int preorderSize, int* inorder, int inorderSize) {
    if(preorderSize<1||preorderSize!=inorderSize)
        return NULL;
    struct TreeNode* root=BuildTree(preorder,0,preorderSize-1,inorder,0,inorderSize-1);
    return root;   
}