本算法和求二叉树的最大深度类似,均可以采用二叉树的层次遍历,遍历一层记录一层元素的数量,然后比较大小即可。
int Width(Bitree bt){
if(bt==NULL) return(0);//空二叉树宽度为0
else{
Bitree Q[];//Q是队列,元素为二叉树结点的指针,容量足够大
front=1;rear=1;last=1;//对头指针、队尾指针、last为最右结点在队列中的位置
temp=0;maxw=0; //temp记局部宽度,maxw记最大宽度
Q[rear]=bt; //根结点入队
while(front<=last){ //循环本层的元素
p=Q[front++];
temp++; //同层元素加一
if(p->lchild!=NUll) Q[rear++]=p->lchild;//左子女入队
if(p->rchild!=NUll) Q[rear++]=p->rchild;//右子女入队
if(front>last){ //一层结束
last=rear; //last指向下一层的最右侧元素
if(temp>maxw) maxw=temp; //更新最大宽度
temp=0; //更新一层的元素数量,方便记录下一层元素数量
}
}
return (maxw);
}
}