两次bfs求树的直径

两次bfs求树的直径

---

方法：先从任意一点R出发，找离它最远的点S，再从点S出发，找离它最远的点T，S到T的距离就是是的直径。
**
**
算法证明:

①若R已经在直径上，根据树的直径的定义可知S也在直径上且为直径的一个端点。T就一定是另一个端点。

②若R不在直径上，我们用反证法，假设此时ST不是直径，AB是直径。
此时的话。证明在RS和AB相交和不相交的情况之下，都由S是树上距离R最远的点可以得到，有比AB更长的路径。与假设相悖，所以，ST就是直径。

//无向图，使用可变长二维数组存树
//第一行输入定点数
//第二行输入与第一个顶点邻接的点个数后面是各个点
#include<iostream>
#include<vector>
#include<cstring>
#include<queue>

using namespace std;
#define N 100

int d1=0;//直径的长度
vector<int> a[N];//记录点和与点有边的点
int bfs(int s)
{
    string color[N];
    int u[N];//前驱
    int d[N];//记录该点到达每一个点的距离
    
    for(int i=0;i<=N;i++)
    {
        color[i]="white";
        u[i]=0;//没有前驱
        d[i]=0;
    }
    color[s]="gray";
    u[s]=0;
    d[s]=0;
    queue<int> q;
    q.push(s);
    while(!q.empty())
    {
        int p=q.front();
        q.pop();
        for(int i=0;i<a[p].size();i++)//遍历邻接矩阵
        {
            int p1=a[p][i];
            if(color[p1]=="white")
            {
                color[p1]="gray";
                d[p1]=d[p]+1;
                u[p1]=p;
                q.push(p1);
            }
        }
        color[p]="black";
    }
    
    for(int i=1;i<N;i++)
    {
        if(d1<d[i])
            d1=d[i];
    }
    int i0=0;
    for(int i=1;i<N;i++)
    {
        if(d1=d[i])
            i0=i;
    }
    return i0;
}
int main()
{
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        int n1;
        cin>>n1;
        for(int j=0;j<=n1;j++)
        {
            int v;
            cin>>v;
            a[i][j]=v;
        }   
    }
    int R=bfs(1);
    int S=bfs(R);//RS就是直径
    cout<<d1<<endl;//d1就是直径的长度
}

（代码没有进行过例子判断）