树的定义改为:
struct mytree
{
int layer;
mytree*left,*right;
ElementType data;
};
新增一个层数layer,可以说明是第几层的,顺序是按照层数从左向右依次进行读取,直到该节点没有左孩子及右孩子为止,同时可以让层序行遍的函数返回一个int类型,layer返回后就是这棵树的层高了。
在其中我们使用STL库里已经定义成功的queue,则有层序行遍的函数
void layerorder(mytree *root)
{queue <mytree*>q;//队列q存储的是tree的指针
root->layer=1;//层数定义为1
q.push(root);//将root的全部地址内容存储至q中
while(!q.empty())
{
mytree *now=q.front();//从队列中取出头指针
q.pop();//释放头指针
cout<<now->data<<endl;
if(now->left!=NULL) //左子树不为空则往下遍历
{
now->left->layer=now->layer+1;
q.push(now->left);
}
if(now->right!=NULL)//同上述操作
{
now->right->layer= now->layer+1;
q.push(now->right);
}