题目大意:
给定二叉树的中序和后序遍历,输出“层次”遍历,这个层次遍历从根节点的子节点开始,先从左到右输出,然后从右到左输出。。。循环往复。
解题思路:
这个题我写的复杂了一点,关于输出哪里我用了两个双端队列模拟了输出的扩展情况,即从左到右和从右到左,所以代码较难理解。维护一个度参数根据奇偶度判断是从右到左还是从左到右输出会方便很多。
代码如下:
#include<iostream>
#include<vector>
#include<queue>
#include<deque>
#include<cstdio>
using namespace std;
int n,post[40],in[40];
struct node
{
int data;
node* lchild;
node* rchild;
};
node* create(int postl,int postr,int inl,int inr)//建树
{
if(postr<postl)return NULL;
node* root=new node;
root->data=post[postr];
int k;
for(int i=inl;i<=inr;i++)
{
if(in[i]==post[postr])
{
k=i;
break;
}
}
int numleft=k-inl-1;
root->lchild=create(postl,postl+numleft,inl,k-1);
root->rchild=create(postl+numleft+1,postr-1,k+1,inr);
return root;
}
void bfs(node* root)//层次遍历
{
deque<node*>q1,q2;//互换扩展节点
queue<int>q3;//保存输出序列
q1.push_back(root);
q3.push(root->data);
int flag=1,num=0;
while(!q1.empty()||!q2.empty())//所有节点都扩展完成
{
while(!q1.empty())//q1是从左到右扩展
{
node* tmp=q1.front();
q1.pop_front();
if(tmp->lchild!=NULL)
{
q2.push_back(tmp->lchild);
node* kk=tmp->lchild;
q3.push(kk->data);
}
if(tmp->rchild!=NULL)
{
q2.push_back(tmp->rchild);
node* kk=tmp->rchild;
q3.push(kk->data);
}
}
while(!q2.empty())//q2是从右到左扩展
{
node* tmp=q2.back();
q2.pop_back();
if(tmp->rchild!=NULL)
{
q1.push_front(tmp->rchild);
node* kk=tmp->rchild;
q3.push(kk->data);
}
if(tmp->lchild!=NULL)
{
q1.push_front(tmp->lchild);
node* kk=tmp->lchild;
q3.push(kk->data);
}
}
}
cout<<q3.front();
q3.pop();
while(!q3.empty())
{
cout<<' '<<q3.front();
q3.pop();
}
cout<<endl;
}
int main()
{
cin>>n;
for(int i=0;i<n;i++)cin>>in[i];
for(int i=0;i<n;i++)cin>>post[i];
node* root=create(0,n-1,0,n-1);
bfs(root);
}