https://pintia.cn/problem-sets/1259041418971803648/problems/1259046259492220928
(题目链接如上)
题意:
给定一棵二叉树的中序遍历和前序遍历,请你先将树做个镜面反转,再输出反转后的层序遍历的序列。所谓镜面反转,是指将所有非叶结点的左右孩子对换。
利用中序遍历和前序遍历建树:
node* buildPreInTree(int pl,int pr,int il,int ir){
if(il>ir) return NULL;
node *root=new node;
root->num=pre[pl];
int mid;
for(int i=il;i<=ir;i++){
if(pre[pl]==in[i]){
mid=i;break;
}
}
int llen=mid-il,rlen=ir-mid;
root->left=buildPreInTree(pl+1,pl+llen,il,mid-1);
root->right=buildPreInTree(pl+llen+1,pr,mid+1,ir);
return root;
}
这道题中,要将root->left和root->right换位,来建树。
二叉树的层序遍历:
int levelOrderTraverse(node* rt){
queue<node*>q;
q.push(rt);
int k=0;
while(!q.empty()){
node* root=q.front();
q.pop();
lev[k++]=root->num;
if(root->left!=NULL) q.push(root->left);
if(root->right!=NULL) q.push(root->right);
}
return k;
}
利用队列,考虑到输出格式,就先把它存储在lev数组里了。
AC代码:
#include <iostream>
#include <queue>
using namespace std;
int n,pre[35],in[35],lev[35];
struct node{
int num;
node *left,*right;
};
node* buildPreInTree(int pl,int pr,int il,int ir){
if(il>ir) return NULL;
node *root=new node;
root->num=pre[pl];
int mid;
for(int i=il;i<=ir;i++){
if(pre[pl]==in[i]){
mid=i;break;
}
}
int llen=mid-il,rlen=ir-mid;
root->right=buildPreInTree(pl+1,pl+llen,il,mid-1);
root->left=buildPreInTree(pl+llen+1,pr,mid+1,ir);
return root;
}
int levelOrderTraverse(node* rt){
queue<node*>q;
q.push(rt);
int k=0;
while(!q.empty()){
node* root=q.front();
q.pop();
lev[k++]=root->num;
if(root->left!=NULL) q.push(root->left);
if(root->right!=NULL) q.push(root->right);
}
return k;
}
int main(){
cin>>n;
for(int i=0;i<n;i++){
cin>>in[i];
}
for(int i=0;i<n;i++){
cin>>pre[i];
}
node *rt;
rt=buildPreInTree(0,n-1,0,n-1);
int k;
k=levelOrderTraverse(rt);
for(int i=0;i<k;i++){
if(i==0) cout<<lev[i];
else cout<<" "<<lev[i];
}
cout<<endl;
return 0;
}