问题描述:给定一个二叉树的中序和后序,给出二叉树的“层次遍历”。
解题思路:首先由中序和后序数组可以直接建树。然后层次遍历,每一层交叉顺序输出。
AC代码:
/*
**1127 ZigZagging on a Tree (30 分)
**层次遍历+中后序建树
**#用层次遍历的思想,设置两个指针,一个指向每一层的开始levbegin,一个指向每一个层结束levend
**#每一层都是逆着遍历父亲节点,然后交替孩子访问顺序。正序则先访问左孩子后右孩子。逆序相反。
*/
#include<iostream>
#include<vector>
using namespace std;
#define MAX 32
typedef struct node{
int val;
struct node *left,*right;
node(int a):val(a),left(NULL),right(NULL){}
}node;
int N,in[MAX],post[MAX];
void creatTree(node *&root,int postR,int inL,int inR)
{
if(inL>inR)return;
int i=inR;
while(i>=0&&postR>=0&&in[i]!=post[postR])--i;//中序中找根节点
root=new node(post[postR]);
creatTree(root->right,postR-1,i+1,inR);
creatTree(root->left,postR-(inR-i)-1,inL,i-1);
}
void printlevel(node *root)
{
vector<node*>sta(N);
if(root){
int f,k,levbegin,levend;
sta[0]=root;
bool turn=1;
node *p;
levbegin=levend=f=0;
while(f<N-1){
k=f+1;//下一层开始指针
if(turn){
while(levend>=levbegin){
p=sta[levend];
if(p->left)sta[++f]=p->left;
if(p->right)sta[++f]=p->right;
--levend;
}
turn=0;
}else{
while(levend>=levbegin){
p=sta[levend];
if(p->right)sta[++f]=p->right;
if(p->left)sta[++f]=p->left;
--levend;
}
turn=1;
}
levend=f;levbegin=k;//本层结束指针
}
printf("%d",root->val);
for(k=1;k<N;++k)printf(" %d",sta[k]->val);
}
}
int main()
{
freopen("test.txt","r",stdin);
int i;
scanf("%d",&N);
for(i=0;i<N;++i)scanf("%d",&in[i]);
for(i=0;i<N;++i)scanf("%d",&post[i]);
node *root=NULL;
creatTree(root,N-1,0,N-1);
printlevel(root);
return 0;
}