HDU 4514 湫湫系列故事——设计风景线 (判环+求最大树直径)

思路：并查集判环，然后bfs求每课树的直径。有个定理，对于一棵树，求它的直径，那么先任意找一点，再求到这个点最远的点为v，然后求到v最远的点的距离就是树的直径。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=100005;
const int maxm=2000005;
int pre[maxn],n,m,res;
int head[maxn],to[maxm],next[maxm],w[maxm],edge;
int tosum[maxn],dis[maxn],que[maxn];
bool vis[maxn],used[maxn];
void init()
{
    edge=0;
    memset(head,-1,sizeof(head));
    memset(used,0,sizeof(used));
    for(int i=1;i<=n;i++)
    pre[i]=i;
}
int find(int x)
{
    int r=x;
    while(r!=pre[r])
        r=pre[r];
    int i=x,j;
    while(i!=r)
    {
        j=pre[i];
        pre[i]=r;
        i=j;
    }
    return r;
}
void unon(int a,int b)
{
    int at = find(a);
    int bt = find(b);
    if(at!=bt)
        pre[at]=bt;
}
inline void add(int u,int v,int c)
{
    to[edge]=v,w[edge]=c,next[edge]=head[u],head[u]=edge++;
    to[edge]=u,w[edge]=c,next[edge]=head[v],head[v]=edge++;
}
void bfs(int s)
{
    memset(vis,0,sizeof(vis));
    memset(dis,0,sizeof(dis));
    int l,r,v,u;
    l=r=0;
    vis[s]=1;
    dis[s]=0;
    que[r++]=s;
    while(r>l)
    {
        u=que[l++];
        used[u]=1;
        for(int i=head[u]; ~i; i=next[i])
            if(!vis[v=to[i]])
            {
                vis[v]=1;
                dis[v]=dis[u]+w[i];
                que[r++]=v;
            }
    }
}
int treediameter(int s)
{
    int u,maxl;
    bfs(s);
    maxl=0,u=s;
    for(int i=1; i<=n; i++)
        if(dis[i]>maxl)
            u=i,maxl=dis[i];
    bfs(u);
    maxl=0;
    for(int i=1; i<=n; i++)
        if(dis[i]>maxl)
            maxl=dis[i];
    return maxl;
}
int main()
{
    int u,v,d,i;
    while(~scanf("%d%d",&n,&m))
    {
        init();
        bool f=0;
        while(m--)
        {
            scanf("%d%d%d",&u,&v,&d);
            if(find(u)==find(v)) f=1;
            unon(u,v);
            add(u,v,d);
        }
        if(f) puts("YES");
        else
        {
            res=-1;
            for(i=1;i<=n;i++)
            if(!used[i]) res=max(res,treediameter(i));
            printf("%d\n",res);
        }
    }
    return 0;
}