解题思路:具体做法就是先随便选一个点,进行bfs或dfs,然后距离这个点最远的点,一定是树直径中的一个端点。然后再从这个端点bfs或dfs求得直径。
证明详见https://www.cnblogs.com/a-clown/p/6131346.html
#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
#define N 10009
using namespace std;
struct node
{
int v,d;
node(int v,int d):v(v),d(d){
}
node(){
}
};
vector<node> s[N];
queue<node> q;
int vis[N],tp;
int bfs(int r)
{
q.push(node(r,0));
memset(vis,0,sizeof(vis));
vis[r]=1;
int k=-1;
tp=0;
while(!q.empty())
{
node t=q.front();
q.pop();
int u=t.v;
if(t.d>tp)
{
tp=t.d;
k=u;
}
for(int i=0;i<s[u].size();i++)
{
int v=s[u][i].v,d=s[u][i].d;
if(!vis[v])
{
q.push(node(v,d+t.d));
vis[v]=1;
}
}
}
return k;
}
int main()
{
//freopen("t.txt","r",stdin);
int a,b,c;
while(scanf("%d%d%d",&a,&b,&c)!=EOF)
{
//printf("%d %d%d\n",a,b,c);
s[a].push_back(node(b,c));
s[b].push_back(node(a,c));
}
int r=bfs(1);
bfs(r);
printf("%d",tp);
return 0;
}