题目链接
题解:通过后序和中序建树,然后层序遍历即可。答案就是每一层树的第一个和最后一个。。
下面是AC代码:
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;
const int N=1010;
int n;
int a[N];
int mid[N];
int last[N];
int tr[N];
vector<int> ans[N];
int maxceng;
void create(int *last,int *mid,int len,int root,int deep)
{
if(len<=0) return ;
int *p;
tr[root]=*(last+len-1);
for(p=mid;p<mid+n;p++)
{
if(*p==*(last+len-1)) break;
}
maxceng=max(maxceng,deep);//找到最大层数
ans[deep].push_back(*(last+len-1));//这里是将每一层的值放到vector里。
int k=p-mid;
create(last,mid,k,root*2,deep+1);//建左子树
create(last+k,p+1,len-k-1,root*2+1,deep+1);//建右子树
}
int main()
{
cin>>n;
memset(tr,-1,sizeof(tr));
for(int i=0;i<n;i++) cin>>mid[i];
for(int i=0;i<n;i++) cin>>last[i];
create(last,mid,n,1,1);
printf("R: ");
for(int i=1;i<=maxceng;i++)
{
int len=ans[i].size();
if(i==maxceng) cout<<ans[i][len-1]<<endl;
else cout<<ans[i][len-1]<<" ";
}
printf("L: ");
for(int i=1;i<=maxceng;i++)
{
if(i==maxceng) cout<<ans[i][0]<<endl;
else cout<<ans[i][0]<<" ";
}
return 0;
}
这里数据量很小,用数组模拟建树即可。