CD 168 二叉树的按层打印与ZIgZag打印

题目描述
给定一颗二叉树，分别实现按层和 ZigZag 打印二叉树。
ZigZag遍历: 意思是第一层从左到右遍历，第二层从右到左遍历，依次类推。
输入描述:
第一行输入两个整数 n 和 root，n 表示二叉树的总节点个数，root 表示二叉树的根节点。
以下 n 行每行三个整数 fa，lch，rch，表示 fa 的左儿子为 lch，右儿子为 rch。(如果 lch 为 0 则表示 fa 没有左儿子，rch同理)
输出描述:
先输出按层打印，再输出按ZigZag打印。
示例1
输入
复制
8 1
1 2 3
2 4 0
4 0 0
3 5 6
5 7 8
6 0 0
7 0 0
8 0 0
输出
复制
Level 1 : 1
Level 2 : 2 3
Level 3 : 4 5 6
Level 4 : 7 8
Level 1 from left to right: 1
Level 2 from right to left: 3 2
Level 3 from left to right: 4 5 6
Level 4 from right to left: 8 7

zlgzag打印时一样按照层序打印的顺序一样，在输出时输入到一个栈中，然后在从栈中输出。

#include<iostream>
#include<queue>
#include<stack>
#include<deque>
using namespace std;

struct node
{
	int lch,rch;
	node(int x=0,int y=0):lch(x),rch(y){};
};
node a[500000];

void printLevel(int root)
{
	queue<int> q;
	q.push(root);
	int level=1;
	while(!q.empty())
	{
		int len=q.size();
		cout<<"Level "<<level++<<" :";
		for(int i=0;i<len;i++)
		{
			int k=q.front();
			q.pop();
			cout<<" "<<k;
			
			if(a[k].lch)
			   q.push(a[k].lch);
			if(a[k].rch)
			   q.push(a[k].rch);
		}
		cout<<endl;
	}
}

void ZigZag(int root)
{
	queue<int> q;
	stack<int> st;
	q.push(root);
	int level=1;
	while(!q.empty())
	{
		
		
		if(level%2==0)//逆序输入到队列中 
		{
			cout<<"Level "<<level<<" from right to left:";
			int len=q.size();
			for(int i=0;i<len;i++)
			{
				int k=q.front();
				q.pop();
				st.push(k);
				if(a[k].lch)
				    q.push(a[k].lch);
				if(a[k].rch )
				    q.push(a[k].rch);
			}
			while(!st.empty())
			{
				cout<<" "<<st.top();
				st.pop();
			}
		}
		else
		{
			cout<<"Level "<<level<<" from left to right:";
			int len=q.size();
			for(int i=0;i<len;i++)
			{
				int k=q.front();
				q.pop();
				cout<<" "<<k;
				if(a[k].lch)
				    q.push(a[k].lch);
				if(a[k].rch )
				    q.push(a[k].rch);
			}
		}
		cout<<endl;	
		level++; 
	}
}


int main()
{
	int n,root;
	scanf("%d%d",&n,&root);
	for(int i=1;i<=n;i++)
	{
		int fa,lch,rch;
		scanf("%d%d%d",&fa,&lch,&rch);
		a[fa].lch=lch;
		a[fa].rch=rch;
	}
 
	printLevel( root);
	ZigZag(root);
	
	return 0;
}