HDOJ-----4514并查集加最长路

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

Time Limit: 6000/3000 MS (Java/Others)    Memory Limit: 65535/32768 K (Java/Others)
Total Submission(s): 4316    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
  
  

首先要说的是这道题，单用并查集是做不出来的，用并查集在根节点储存最长距离错误的，因为题目并没有说树没有枝杈，而且并不是只有一棵树，所以要遍历所有的边才能找到所有树中的最长路

然后就是并查集判断是否有环，再最长路计算就可以了，稍长了点，有段时间没写百行的代码了，不过也就是长了点，其他就没啥了

#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
#define maxn 1000010 
using namespace std;
int ans, num, ok, nod, s, e, cnt;
int dis[maxn], pre[maxn], head[maxn], sum[maxn];
bool vis[maxn], visi[maxn];
int find(int a){
	return a == pre[a] ? a : pre[a] = find(pre[a]);
}
void merge(int x, int y){
	int fx = find(x), fy = find(y);
	if(fx != fy){
		pre[fx] = fy;
		return ;
	}
	ok++;
}
struct node{
	int from, to, val, next;
}edge[maxn*2];
void add(int u, int v, int w){
	edge[num].from = u;
	edge[num].to = v;
	edge[num].val = w;
	edge[num].next = head[u];
	head[u] = num++;
}
void bfs(int t){
	memset(vis, false, sizeof(vis));
	memset(dis, 0, sizeof(dis));
	queue<int > Q;
	Q.push(t);
	vis[t] = true;
	ans = 0;
	while(!Q.empty()){
		int u = Q.front();
		Q.pop();
		visi[u] = true;
		for(int i = head[u]; i != -1; i = edge[i].next){
			int v = edge[i].to;
			if(!vis[v]){
				dis[v] = edge[i].val + dis[u];
				vis[v] = true;
				Q.push(v);
				if(ans < dis[v]){
					ans = dis[v];
					nod = v;
				}
			}
		}
	}
}
int main(){
	int m, n, a, b, c;
	while(~scanf("%d%d", &m, &n)){
		memset(head, -1, sizeof(head));
		memset(visi, false, sizeof(visi));
		memset(pre, 0, sizeof(pre));
		memset(visi, false, sizeof(visi));
		for(int i = 1; i <= m; i++){//本来想动态建立的，不知道为什么行不通
			pre[i] = i;
		}
		num = ok = cnt = 0;
		for(int i = 0; i < n; i++){
			scanf("%d%d%d", &a, &b, &c);
			if(ok){
				continue;
			}
			merge(a, b);
			add(a, b, c);
			add(b, a, c);
		}
		if(ok || n >= m){//m个点，m-1条边即可连通，多余的肯定成环，n>=m这个判断貌似应该再靠前一些
			printf("YES\n");
			continue;
		}
		for(int i = 1; i <= m; i++){
			if(!visi[i]){
				bfs(i);
				bfs(nod);
			}
			if(cnt < ans){
				cnt = ans;
			}
		}
		printf("%d\n", cnt);
	}
	return 0;
}