889. 根据前序和后序遍历构造二叉树
给定两个整数数组,preorder
和 postorder
,其中 preorder
是一个具有 无重复 值的二叉树的前序遍历,postorder
是同一棵树的后序遍历,重构并返回二叉树。
如果存在多个答案,您可以返回其中 任何 一个。
示例 1:
输入:preorder = [1,2,4,5,3,6,7], postorder = [4,5,2,6,7,3,1]
输出:[1,2,3,4,5,6,7]
示例 2:
输入: preorder = [1], postorder = [1]
输出: [1]
提示:
1 <= preorder.length <= 30
1 <= preorder[i] <= preorder.length
preorder
中所有值都 不同postorder.length == preorder.length
1 <= postorder[i] <= postorder.length
postorder
中所有值都 不同- 保证
preorder
和postorder
是同一棵二叉树的前序遍历和后序遍历
题解
先序和中序,中序和后序都可以确定一棵树,但是先序和后序确定的树不唯一,题目要求输出一个即可。
一定要先做前两题105. 从前序与中序遍历序列构造二叉树和106. 从中序与后序遍历序列构造二叉树。
依旧是采用递归思想,根据示例一,[1,2,4,5,3,6,7],[4,5,2,6,7,3,1]
,可以发现当preorder
下标为cnt=0
时,pre[cnt]=1
,对应的post
中为post[i=6]
,
接下来分析,针对节点1
,如果pre
中右边一个的节点2
与post
中左边一个的节点3
不一样,其实代表2
不是1
的右子树(否则会一样),那就直接把它分配给左子树。
那么首先针对pre
中的节点需要在post
中遍历找到其对应的节点,然后判断pre
右边一个的节点是属于左子树还是右子树。然后建树,就可以得到一个符合题目要求的答案(注意答案不唯一)。
查看代码
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
int cnt=0;//计数遍历到preorder的哪个下标
int len=0;//节点总数
int flag[3000];//是否访问过的标志位
TreeNode * build(TreeNode*& cur,vector<int>& postorder,vector<int>& preorder){
int val=preorder[cnt++];//获取节点值
int i;
for(i=0;i<len;i++)//在postorder中寻找节点
if(postorder[i]==val)
break;
flag[i]=1;//标注此节点被访问
cur=new TreeNode(val);//新建节点
if(cnt<len&&i>0&&i<len&&preorder[cnt]!=postorder[i-1]&&flag[i-1]!=1)//如果不相等就分给左子树
cur->left=build(cur->left,postorder,preorder);
if(cnt<len&&i>0&&i<len&&preorder[cnt]==postorder[i-1]&&flag[i-1]!=1)//否则右子树
cur->right=build(cur->right,postorder,preorder);
return cur;//返回建好左右子树的当前节点
}
TreeNode* constructFromPrePost(vector<int>& preorder, vector<int>& postorder) {
TreeNode * root;
len=postorder.size();
memset(flag,0,sizeof(len));//初始化len长度为0
root=build(root,postorder,preorder);
return root;
}
};