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<iostream>
#include<queue>
using namespace std;
#define N 50
int pre[N],in[N];
int n;
struct node{
int date;
node* l;
node* r;
};
node* create(int preL,int preR,int inL,int inR){
if(preL>preR) return NULL;
node *root= new node;
root->date = pre[preL];
int k;
for(k=inL;k<=inR;k++)
if(in[k]==pre[preL])
break;
int numL = k-inL;
root->l = create(preL+1,preL+numL,inL,k-1);
root->r = create(preL+numL+1,preR,k+1,inR);
return root;
}
int flag=0;
void bfs(node* root){
queue<node*> q;
q.push(root);
while(q.size()){
node* now =q.front();
q.pop();
if(flag)cout<<" "<<now->date;
else{
cout<<now->date;
flag=1;
}
if(now->r!=NULL) q.push(now->r);
if(now->l!=NULL) q.push(now->l);
}
}
int main(){
cin>>n;
int i;
for(i=0;i<n;i++)
cin>>in[i];
for(i=0;i<n;i++)
cin>>pre[i];
node* root = create(0,n-1,0,n-1);
bfs(root);
}