leetcode系列106-从中序与后序遍历序列构造二叉树

最新推荐文章于 2021-01-09 19:08:02 发布

HanWagner

最新推荐文章于 2021-01-09 19:08:02 发布

阅读量166

点赞数

分类专栏： LeetCode 文章标签：二叉树 leetcode 数据结构

本文链接：https://blog.youkuaiyun.com/qq_30286973/article/details/107532655

版权

LeetCode 专栏收录该内容

96 篇文章

订阅专栏

【题目概要】

106. 从中序与后序遍历序列构造二叉树
根据一棵树的中序遍历与后序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。

例如，给出
中序遍历 inorder = [9,3,15,20,7]
后序遍历 postorder = [9,15,7,20,3]
返回如下的二叉树：
    3
   / \
  9  20
    /  \
   15   7

【代码示例】

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */

struct TreeNode* buildTree(int* inorder, int inorderSize, int* postorder, int postorderSize){
    if(!inorder || !postorder)
    return NULL;
    if(inorderSize == 0 || postorderSize == 0)
    return NULL;
    // if(inorderSize != postorderSize)
    // return NULL;
    struct TreeNode *root = (struct TreeNode *)malloc(sizeof(struct TreeNode));
    root->val = postorder[postorderSize-1];
    int leftnum = 0, rightnum = 0;
    // for(int count=0; count<inorderSize; count++)
    // {
    //     if(inorder[count] == postorder[postorderSize-1])
    //     break;
    //     leftnum++;
    // }
    while(leftnum<inorderSize && inorder[leftnum] != postorder[postorderSize-1])
    leftnum++;
    rightnum = inorderSize-leftnum-1;
    root->left = buildTree(inorder, leftnum, postorder, leftnum);
    root->right = buildTree(inorder+leftnum+1, rightnum, postorder+leftnum, rightnum);
    return root;
}