题目
leetcode 106
根据一棵树的中序遍历与后序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如,给出
中序遍历 inorder = [9,3,15,20,7]
后序遍历 postorder = [9,15,7,20,3]
返回如下的二叉树:
代码
后续遍历最后一个节点是根节点,中序遍历按根节点分为左右部分
//后续遍历最后一个节点是根节点,中序遍历按根节点分为左右部分
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 *inPostbuild(int *inorder, int l1, int r1, int *postorder, int l2, int r2) {
if (l1 > r1) return NULL;
struct TreeNode *node = (struct TreeNode *)malloc(sizeof(struct TreeNode));
int bit = local(inorder, l1, r1, postorder[r2]);
node->val = postorder[r2];
node->left = inPostbuild(inorder, l1, bit - 1, postorder, l2, l2 + bit - l1 - 1);
node->right = inPostbuild(inorder, bit + 1, r1, postorder, l2 + bit - l1, r2 - 1);
return node;
}
struct TreeNode* buildTree(int* inorder, int inorderSize, int* postorder, int postorderSize){
struct TreeNode *head;
head = inPostbuild(inorder, 0, inorderSize - 1, postorder, 0, postorderSize - 1);
return head;
}