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 3
1 2 1
2 3 1
3 1 1

 

Sample Output

YES今天刚学的最长直径问题，所以一刚拿到题就想着用bfs找最长直径，中间再加一点并查集，判断是否有环生成，，呼呼的写了半天交上去后一直是爆栈，哎。。心塞。。然后经过一系列的改进后又表示答案错误，这已经没shei了，智商不够用，没办法。。。思路：先用并查集判断是否有环生成，如果没有的话就用两次bfs查找最长路径即可，具体见如下代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>
#define maxn 1000010
using 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 4
1 2 2
1 3 4
4 5 4
4 6 3

*/ // 用bfs做的，这组数据得注意点