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

题目描述:

 

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

 

代码如下:

/**
 * 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;   
}