两次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就是直径的长度
}
(代码没有进行过例子判断)