leetcode 105 从前序与中序遍历序列构造二叉树 C语言实现

题目

leetcode 105
根据一棵树的前序遍历与中序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如,给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树:
leetcode 105 从前序与中序遍历序列构造二叉树 C语言实现

代码

前序遍历的第一个节点作为根节点,把中序遍历分为根节点的左边节点和右边节点,递归实现

int local(int *nums, int l, int r, int x) {
    for (int i = l; i <= r; i++) {
        if (nums[i] == x) return i;
    }
    return -1;
}
//前序遍历的第一个节点作为根节点,把中序遍历分为根节点的左边节点和右边节点,递归实现
struct TreeNode *preInorder(int *preorder, int l1, int r1, int *inorder, int l2, int r2) {
    if (l1 > r1) return NULL;
    struct TreeNode *node = (struct TreeNode *)malloc(sizeof(struct TreeNode));
    int bit = local(inorder, l2, r2, preorder[l1]);
    node->val = preorder[l1];
    node->left = preInorder(preorder, l1 + 1, l1 + bit - l2, inorder, l2, bit - 1);
    node->right = preInorder(preorder, l1 + bit - l2 + 1, r1, inorder, bit + 1, r2);
    return node;
}
struct TreeNode* buildTree(int* preorder, int preorderSize, int* inorder, int inorderSize){
    struct TreeNode *head;
    head = preInorder(preorder, 0, preorderSize - 1, inorder, 0, inorderSize - 1);
    return head;
}
leetcode 105 从前序与中序遍历序列构造二叉树 C语言实现leetcode 105 从前序与中序遍历序列构造二叉树 C语言实现 瓶邪写一个夏天 发布了64 篇原创文章 · 获赞 2 · 访问量 2902 私信 关注
上一篇:刷题105. Construct Binary Tree from Preorder and Inorder Traversal


下一篇:A1064 Complete Binary Search Tree (30分)