题目
https://pintia.cn/problem-sets/994805342720868352/problems/994805345078067200
题意
给出一颗二叉树的前序和中序序列,输出后序序列的第一个树
题目解析
要求输出的这个点其实就是沿着二叉树根节点一路向左寻找的第一个叶子节点(后序序列的顺序为左右根,所以必须是叶子节点),所以我们就通过递归沿着中序序列的根节点所在位置不断向左寻找即可
注意
如果没有左子树,那么直接寻找右子树即可,这样也能节省时间
AC代码
#include<bits/stdc++.h>
using namespace std;
int n,pre[50001],in[50001];
void find(int root,int l,int r)
{
if(l==r)
{
cout<<pre[root];
return;
}
int pos=l;
while(in[pos]!=pre[root]) pos++;
if(pos==l) find(root+1+(pos-l),l+1,r);
else find(root+1,l,pos-1);
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
cin>>pre[i];
for(int i=1;i<=n;i++)
cin>>in[i];
find(1,1,n);
}