面试题_分层遍历二叉树
编程之美上的题目。
问题1:给定一棵二叉树,要求按分层遍历该二叉树,即从上到下按层次访问该二叉树(每一行将单输出一行),每一层要求访问的顺序为从左向右,并将节点依次编号。
问题2:写一个函数,打印二叉树中某层次的节点(从左向右),其中根结点为第1层。
#include<iostream>
#include<queue>
using namespace std;
struct Tnode
{
Tnode * leftChild;
Tnode * rightChild;
int value;
Tnode():leftChild(NULL),rightChild(NULL)
{
value=0;
}
Tnode(int i):leftChild(NULL),rightChild(NULL)
{
value=i;
}
};
//代码之美上的。输出第level层的所有元素
void printNodeAtLevel(Tnode * root, int level)
{
if(root==NULL||level<0)
{
return ;
}
if(level==1)
{
cout<<root->value<<" ";
return ;
}
printNodeAtLevel(root->leftChild,level-1);
printNodeAtLevel(root->rightChild,level-1);
}
int main()
{
//建树
Tnode * node1=new Tnode(1);
Tnode * node2= new Tnode(2);
Tnode * node3= new Tnode(3);
Tnode * node4= new Tnode(4);
Tnode * node5= new Tnode(5);
Tnode * node6= new Tnode(6);
Tnode * node7= new Tnode(7);
Tnode * node8= new Tnode(8);
node1->leftChild=node2;
node1->rightChild=node3;
node2->leftChild=node4;
node2->rightChild=node5;
node3->rightChild=node6;
node5->leftChild=node7;
node5->rightChild=node8;
//按层序遍历二叉树,并在每一层所有元素的后面打印一个换行符
//编程之美上写的是类似递归的思想,其实也是在用递归栈来记录层数,
//那么不如自己用个队列来实时追踪当前元素所在层
queue<Tnode*> que;
queue<int> level;//用一个辅助队列记录当前元素的所在层
que.push(node1);
level.push(1);
int currentLevel=1;
while(!que.empty())
{
Tnode * temp=que.front();
que.pop();
int l=level.front();
level.pop();
if(currentLevel<l)
{
cout<<endl;
currentLevel=l;
}
cout<<temp->value<<" ";
if(temp->leftChild!=NULL)
{
que.push(temp->leftChild);
level.push(l+1);
}
if(temp->rightChild!=NULL)
{
que.push(temp->rightChild);
level.push(l+1);
}
}
cout<<endl;
// printNodeAtLevel(node1,3);
return 0;
}