团体程序设计天梯赛-练习集L2-011. 玩转二叉树

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 ;
 }