考研数据结构(每日一题)
题目:假设二叉树采用二叉链表存储结构,求非空二叉树b的宽度(即具有结点数最多的那一层的结点个数)
算法思想:
采用层次遍历的方法。
首先求出所有结点的层次,将所有结点和对应的层次放在一个队列中。
然后依次扫描队列,并求出各层的结点总数,最大的层结点总数即为二叉树的宽度。
完整代码:
typedef struct{
BiTree data[MaxSize]; //保存队列中的结点指针
int level[MaxSize]; //保存data中相同下标结点的层次
int front,rear;
}Qu;
int BTWidth(BiTree b){
BiTree p;
int k,max,i,n;
Qu.front = Qu.rear = -1; //队列为空
Qu.rear ++;
Qu.data[Qu.rear] = b; //根结点指针入队
Qu.level[Qu.rear] = 1; //根结点的层次为1
while (Qu.front < Qu.rear)
{
Qu.front ++; //出队
p = Qu.data[Qu.front]; //出队结点
k = Qu.level[Qu.front]; //出队结点的层次
if (p -> lchild != NULL) //左孩子入队
{
Qu.rear ++;
Qu.data[Qu.rear] = p -> lchild;
Qu.level[Qu.rear] = K + 1;
}
if (p -> rchild != NULL){ //右孩子入队
Qu.rear ++;
Qu.data[Qu.rear] = p -> rchild;
Qu.level[Qu.rear] = K + 1;
}
}
max = 0; //max保存同一层最多的结点个数
i = 0;
k = 1; //k表示从第一层开始查找
while(i <= Qu.rear){ //i扫描队中所有元素
n = 0; //n统计第k层的结点个数
while(i <= Qu.rear && Qu.level[i] == k){
n ++;
i ++;
}
k = Qu.level[i];
if(n > max){
max = n; //保存最大的n
}
}
return max;
}