18924 二叉树的宽度

Description
二叉树的宽度指的是具有节点数目最多的那一层的节点个数。
1
/
2 3
/
4
答案为2, 第二层节点数最多，为2个节点。

输入格式
共n行。
第一行一个整数n，表示有n个结点，编号为1至n,结点1为树根。（1<=n<=50）
第二行至第n行，每行有两个整数x和y，表示在二叉树中x为y的父节点。x第一次出现时y为左孩子

输出格式
输出二叉树的宽度。

输入样例
5
1 2
1 3
2 4
2 5

输出样例
2

思路：

二叉树的顺序存储结构、广搜

代码：

#include <cstdio>
#include <iostream>
using namespace std;

typedef struct BiTNode
{
    int data;
    int lchild, rchild;
}BiTNode;

int bfs(BiTNode * T)
{
    int sum = 0;
    int q[10000] = {0}, f = 0, r = 0;//队列存储编号
    q[r++] = 1;//第一个节点编号入队
    while (f!=r)
    {
        int len = r - f, i;//当前层的宽度
        sum = sum > len ? sum : len;//统计最大宽度
        for (i = 0; i < len; i++)
        {
            int headIndex = q[f++];//队头结点编号出队
            if (T[headIndex].lchild) q[r++] = T[headIndex].lchild;//左孩子节点编号入队
            if (T[headIndex].rchild) q[r++] = T[headIndex].rchild;//右孩子节点编号入队
        }
    }
    return sum;//返回最大宽度
}

int main()
{
    BiTNode T[55] = {0};
    int n,x,y,i;
    cin>>n;
    for (i = 1;i < n; i++)//n个节点，存储n-1条节点和节点关系
    {
        cin>>x>>y;
        if (T[x].lchild == 0)//如果x没有左孩子
            T[x].lchild = y;
        else//如果x有左孩子
            T[x].rchild = y;
    }
    cout<<bfs(T)<<endl;
    return 0;
}