树的操作--层序遍历(3)

来源:互联网 发布:飞利浦脱毛仪 知乎 编辑:程序博客网 时间:2024/06/15 04:32

树的定义改为:

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);
}
原创粉丝点击