【HDU OJ 4514】湫湫系列故事——设计风景线

湫湫系列故事——设计风景线

Problem Description
　　随着杭州西湖的知名度的进一步提升，园林规划专家湫湫希望设计出一条新的经典观光线路，根据老板马小腾的指示，新的风景线最好能建成环形，如果没有条件建成环形，那就建的越长越好。
　　现在已经勘探确定了n个位置可以用来建设，在它们之间也勘探确定了m条可以设计的路线以及他们的长度。请问是否能够建成环形的风景线？如果不能，风景线最长能够达到多少？
　　其中，可以兴建的路线均是双向的，他们之间的长度均大于0。

Input
　　测试数据有多组，每组测试数据的第一行有两个数字n, m，其含义参见题目描述；
　　接下去m行，每行3个数字u v w，分别代表这条线路的起点，终点和长度。

　　[Technical Specification]
　　1. n<=100000
　　2. m <= 1000000
　　3. 1<= u, v <= n
　　4. w <= 1000

Output
　　对于每组测试数据，如果能够建成环形（并不需要连接上去全部的风景点），那么输出YES，否则输出最长的长度，每组数据输出一行。

Sample Input
3 3
1 2 1
2 3 1
3 1 1

Sample Output
YES

#include<cstdio>#include<cstring>#include<queue>#include<algorithm>#define K 100011using namespace std;struct node{    int fo,to,val,next;}st[K*20];bool vis[K],H;int dis[K],num;int head[K];int bo[K];int N,M,ans,cot;void add(int x,int y,int z){    node E={x,y,z,head[x]};    st[num]=E;    head[x]=num++;}void BFS(int fx){    queue <int> q;    while(!q.empty())    q.pop();    memset(dis,0,sizeof(dis));    memset(vis,false,sizeof(vis));    vis[fx]=true;    ans=0;    cot=fx;    q.push(fx);    while(!q.empty())    {        int w=q.front();        q.pop();        for(int i=head[w];i!=-1;i=st[i].next)        {           node E=st[i];           if(!vis[E.to]&&dis[E.to]<E.val+dis[w])           {            vis[E.to]=true;            dis[E.to]=E.val+dis[w];             if(dis[E.to]>ans)            {                ans=dis[E.to];                cot=E.to;            }            q.push(E.to);          }        }    }}int find(int p){    int ch=p;    int t;    while(p!=bo[p])    p=bo[p];    return p;}void KL(int x,int y){    int fx=find(x);    int fy=find(y);    if(fx!=fy)    bo[fy]=fx;    else    H=true;}int main(){    int i,x,y,z,o;    while(scanf("%d%d",&N,&M)!=EOF)    {        num=0;        for(i=0;i<=N;i++)        bo[i]=i;        H=false;        memset(head,-1,sizeof(head));        for(i=1;i<=M;i++)        {            scanf("%d%d%d",&x,&y,&z);            o=x;            if(H)            continue;            KL(x,y);            add(x,y,z);            add(y,x,z);        }        if(H)        {            printf("YES\n");            continue;        }            BFS(o);            BFS(cot);            printf("%d\n",ans);    }    return 0;}