hdu4514 湫湫系列故事——设计风景线(坑，树不止一个）

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

Time Limit: 6000/3000 MS (Java/Others)    Memory Limit: 65535/32768 K (Java/Others)
Total Submission(s): 4040    Accepted Submission(s): 722

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 <stdio.h>
#include <queue>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn=1100010;
int f[100010],n,m,num,head[100010],dis[100010];
int used[100010],vis[100010];
struct EDGE{
    int to,w,next;
}edge[maxn*2];
void addedge(int u,int v,int w){//构建链表
    edge[num].to=v;
    edge[num].w=w;
    edge[num].next=head[u];
    head[u]=num++;
}
void bfs(int s){
    memset(dis,0,sizeof(dis));
    memset(vis,0,sizeof(vis));
    vis[s]=1;dis[s]=0;
    queue<int>que;
    que.push(s);
    while(!que.empty()){
        int u=que.front();que.pop();
        used[u]=1;
        for(int i=head[u];i!=-1;i=edge[i].next){
            int v=edge[i].to;
            if(vis[v]==0){
                vis[v]=1;
                dis[v]=dis[u]+edge[i].w;
                que.push(v);//直接记录，因为不只一棵树
            }
        }
    }
}
int slove(int s){
    int max1=0,pos=s;
    bfs(s);
    for(int i=1;i<=n;i++){
        if(dis[i]>max1){
            pos=i;max1=dis[i];
        }
    }
    bfs(pos);
    int ans=0;
    for(int i=1;i<=n;i++){
        if(dis[i]>ans)
 ans=dis[i];
    }
    return ans;//找出最大分支
}
void init(){
    for(int i=0;i<=n;i++) f[i]=i;
    num=0;
    memset(head,-1,sizeof(head));
    memset(used,0,sizeof(used));
}
int find1(int x){//要压缩路径，那不然超时
    int k,r=x,j;
    while(r!=f[r]) r=f[r];
    k=x;
    while(k!=r){
        j=f[k];
        f[k]=r;k=j;
    }
    return r;
}
void unite(int a,int b,int c){
    int aa=find1(a);
    int bb=find1(b);
    f[aa]=bb;
}
int main(){
    int a,b,c;
    while(scanf("%d%d",&n,&m)!=-1){
        init();
        int flag=0;
        for(int i=0;i<m;i++){
            scanf("%d%d%d",&a,&b,&c);
            if(find1(a)==find1(b))
           flag=1;
            unite(a,b,c);
            addedge(a,b,c);addedge(b,a,c);
        }
        if(flag) printf("YES\n")//有回路;
        else{
            int ans=-1;
            for(int i=1;i<=n;i++){
                if(used[i]==0){
                    ans=max(ans,slove(i));//找出最长分支
                }
            }
            printf("%d\n",ans);
        }
    }
    return 0;
}