106.根据一棵树的中序遍历与后序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
- 我们都知道后序遍历最后一个元素就是根结点,所以可以将后序遍历最后一个元素取出来,然后利用这个元素去分割中序遍历,得到中序遍历中二叉树左子树和右子树的元素所在范围,并且根据元素个数还能够得到后序遍历中二叉树左子树和右子树的元素所在范围。于是我们就有了二叉树左子树的中序遍历和后序遍历以及右子树的中序遍历和后序遍历,递归地对左右子树进行处理即可构造出所要的二叉树。每次对中序遍历序列和后序遍历序列进行划分后,都会产生四个区间,在递归函数传递区间的范围的时候,要坚持某种区间的划分规则,比如我用的是左闭右闭的规则,即传递到递归函数中的下标都是可以在数组上取到值的。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
//i,j来界定进入递归的中序序列,m,n用来界定进入递归的后序序列
struct TreeNode* build(int* inorder, int i, int j, int* postorder, int m, int n) {
if (i > j)
return NULL;
struct TreeNode* root = malloc(sizeof(struct TreeNode));
root->val = postorder[n];
int index = 0;
for (int k = i; k <= j; k++) {
if (root->val == inorder[k]) {
index = k;
break;
}
}
root->left = build(inorder, i, index - 1, postorder, m, index - i + m - 1);
root->right = build(inorder, index + 1, j, postorder, index - i + m, j - i + m - 1);
return root;
}
struct TreeNode* buildTree(int* inorder, int inorderSize, int* postorder, int postorderSize){
if (inorderSize == 0 || postorderSize == 0)
return NULL;
return build(inorder, 0, inorderSize - 1, postorder, 0, postorderSize - 1);
}
105.给定一棵树的前序遍历 preorder 与中序遍历 inorder。请构造二叉树并返回其根节点。
- 105题跟106题基本是一模一样的,只不过取根节点的地方变成在前序遍历序列的首个元素取,然后区间划分稍微改动一下。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
//i\j来界定前序遍历序列,m、n来界定中序遍历序列
struct TreeNode* build(int* preorder, int i, int j, int* inorder, int m, int n) {
if (i > j)
return NULL;
struct TreeNode* root = malloc(sizeof(struct TreeNode));
root->val = preorder[i];
int index = 0;
for (int k = m; k <= n; k++) {
if (root->val == inorder[k]) {
index = k;
break;
}
}
root->left = build(preorder, i + 1, index - m + i, inorder, m, index - 1);
root->right = build(preorder, j - n + index + 1, j, inorder, index + 1, n);
return root;
}
struct TreeNode* buildTree(int* preorder, int preorderSize, int* inorder, int inorderSize){
if (preorderSize == 0 || inorderSize == 0)
return NULL;
return build(preorder, 0, preorderSize - 1, inorder, 0, inorderSize - 1);
}