hd 4514 湫湫系列故事——设计风景线（并查集+最长直径）

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

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

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 31 2 12 3 13 1 1

 

Sample Output

YES
今天刚学的最长直径问题，所以一刚拿到题就想着用bfs找最长直径，中间再加一点并查集，判断是否有环生成，，呼呼的写了半天交上去后一直是爆栈，哎。。心塞。。然后经过一系列的改进后又表示答案错误，这已经没shei了，智商不够用，没办法。。。
思路：先用并查集判断是否有环生成，如果没有的话就用两次bfs查找最长路径即可，具体见如下代码：
#include<cstdio>#include<cstring>#include<iostream>#include<queue>#define maxn 1000010using namespace std;int pre[maxn];struct Edge{  int from;  int to;  int val;  int next;};bool falg;int find(int x){    if(x != pre[x])       pre[x] = find(pre[x]);    return pre[x];}Edge edge[maxn * 2];int edgenum;int head[maxn];int dist[maxn];bool vis[maxn];int tnode,ans;void init(){    memset(head,-1,sizeof(head));    edgenum = 0;}void addEdge(int u,int v,int w){    edge[edgenum].to = v;    edge[edgenum].val = w;    edge[edgenum].next = head[u];    head[u] = edgenum++;}void bfs(int s){    memset(vis,false,sizeof(vis));    memset(dist,0,sizeof(dist));    queue<int > q;    q.push(s);    vis[s] = true;    ans = 0;    dist[s] = 0;    while(!q.empty())    {        int u = q.front();        q.pop();        for(int i = head[u] ; i != -1 ; i = edge[i].next)        {            int v = edge[i].to;            if(!vis[v])            {                if(dist[v] < dist[u] + edge[i].val)                    dist[v] = dist[u] + edge[i].val;                vis[v] = true;                q.push(v);                if(ans < dist[v])                {                   ans = dist[v];                   tnode = v;                 }            }        }    }}int main(){    int n,m,a,b,c;    while(~scanf("%d%d",&n,&m))    {        init();        falg = true;        for(int i = 1 ; i <= n ; i++)            pre[i] = i;        for(int i = 0 ; i < m ; i++)        {            scanf("%d%d%d",&a,&b,&c);            int fx = find(a);            int fy = find(b);            if(fx != fy)            {                pre[fx] = fy;                addEdge(a,b,c);                addEdge(b,a,c);            }             else                 falg = false;        }        if(falg == false)            printf("YES\n");        else        {            int sum = 0;            for(int i = 1; i <= n ; i++)  // 判断这个图中可能会存在多颗树，如下边那组数据，所以需遍历一次才能找出最长路径               {                if(pre[i] == i)                {                     bfs(i);bfs(tnode);                     sum = max(sum,ans);                }            }             printf("%d\n",sum);        }    }    return 0;}
/*
6 41 2 21 3 44 5 44 6 3
*/   // 用bfs做的，这组数据得注意点