L2-011 玩转二叉树 (25分)
给定一棵二叉树的中序遍历和前序遍历,请你先将树做个镜面反转,再输出反转后的层序遍历的序列。所谓镜面反转,是指将所有非叶结点的左右孩子对换。这里假设键值都是互不相等的正整数。
输入格式:
输入第一行给出一个正整数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
本题先按照常规方法建树,然后输出的时候按照先序的镜像进行输出即可
//@author:hairu,wu
//@from:ahut
#include<iostream>
#include<vector>
#include<queue>
using namespace std;
const int max_n=1e3;
int in[max_n];
int pre[max_n];
struct Tree{
int data;
Tree *lchild;
Tree *rchild;
Tree(){
lchild=NULL;
rchild=NULL;
}
};
void createTree(Tree* &tree,int abegin,int aend,int bbegin,int bend){
if(abegin > aend) return ;
if(tree==NULL){
tree=new Tree();
tree->data=pre[bbegin];
}
int i;
for(i=abegin;i!=aend;i++){
if(in[i]==pre[bbegin]){
break;
}
}
createTree(tree->lchild,abegin,i-1,bbegin+1,bbegin+(i-abegin));
createTree(tree->rchild,i+1,aend,bbegin+(i-abegin)+1,bend);
}
void lastorder(Tree* tree){
if(tree == NULL){
return ;
}
lastorder(tree->lchild);
lastorder(tree->rchild);
cout<<tree->data<<" ";
}
void preorder(Tree *tree){
if(tree==NULL){
return ;
}
cout<<tree->data<<" ";
preorder(tree->lchild);
preorder(tree->rchild);
}
vector<int> levelorder(Tree *t){
vector<int> ans;
queue<Tree *> q;
if(t!=NULL){
q.push(t);
}
while(!q.empty()){
Tree* node=q.front();
q.pop();
ans.push_back(node->data);
if(node->rchild!=NULL){
q.push(node->rchild);
}
if(node->lchild!=NULL){
q.push(node->lchild);
}
}
return ans;
}
int main(){
int n;
cin >> n;
for(int i=0;i<n;i++){
cin >> in[i];
}
for(int i=0;i<n;i++){
cin >> pre[i];
}
//构造二叉树
Tree *tree=NULL;
createTree(tree,0,n-1,0,n-1);
vector<int> ans=levelorder(tree);
int size=ans.size();
for(int i=0;i<size;i++){
cout<<ans[i];
if(i!=size-1) cout<<" ";
else cout<<endl;
}
return 0;
}