L2-011. 玩转二叉树
时间限制
400 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
作者
陈越
给定一棵二叉树的中序遍历和前序遍历，请你先将树做个镜面反转，再输出反转后的层序遍历的序列。所谓镜面反转，是指将所有非叶结点的左右孩子对换。这里假设键值都是互不相等的正整数。
输入格式：
输入第一行给出一个正整数N（<=30），是二叉树中结点的个数。第二行给出其中序遍历序列。第三行给出其前序遍历序列。数字间以空格分隔。
输出格式：
在一行中输出该树反转后的层序遍历的序列。数字间以1个空格分隔，行首尾不得有多余空格。
输入样例：
7
1 2 3 4 5 6 7
4 1 3 2 6 5 7
输出样例：
4 6 1 7 5 3 2
思路：跟树的遍历一样
#include<bits/stdc++.h>
using namespace std;
#define maxn 1000
struct Node{
int l;
int r;
}aa[maxn];
int f[maxn],m[maxn];
int build(int la,int ra,int lb,int rb){
if(la>ra)
return ;
int root,p1,p2;
root=f[lb];
p1=la;
while(m[p1]!=root) p1++;
p2=p1-la;
aa[root].l=build(la,p1-,lb+,lb+p2);
aa[root].r=build(p1+,ra,lb+p2+,rb);
return root;
}
void bfs(int root){
queue<int> q;
vector<int> v;
q.push(root);
while(!q.empty()){
int w=q.front();
q.pop();
if(w==)
break;
v.push_back(w);
if(aa[w].r!=) q.push(aa[w].r);
if(aa[w].l!=) q.push(aa[w].l);
}
int len=v.size();
for(int i=;i<len;i++)
printf("%d%c",v[i],i==len-?'\n':' ');
}
int main(){
int n;
cin>>n;
for(int i=;i<n;i++) cin>>m[i];
for(int i=;i<n;i++) cin>>f[i];
build(,n-,,n-);
int root=f[];
bfs(root);
return ;
}