1882: 修路方案 NYOJ 118

1882: 修路方案

时间限制: 3 Sec  内存限制: 64 MB
提交: 43  解决: 18
您该题的状态：已完成
[提交][状态][讨论版]

题目描述

南将军率领着许多部队，它们分别驻扎在N个不同的城市里，这些城市分别编号1~N，由于交通不太便利，南将军准备修路。

现在已经知道哪些城市之间可以修路，如果修路，花费是多少。

现在，军师小工已经找到了一种修路的方案，能够使各个城市都联通起来，而且花费最少。

但是，南将军说，这个修路方案所拼成的图案很不吉利，想让小工计算一下是否存在另外一种方案花费和刚才的方案一样，现在你来帮小工写一个程序算一下吧。

输入

第一行输入一个整数T(1<T<20)，表示测试数据的组数
每组测试数据的第一行是两个整数V,E，(3<V<500,10<E<200000)分别表示城市的个数和城市之间路的条数。数据保证所有的城市都有路相连。
随后的E行，每行有三个数字A B L，表示A号城市与B号城市之间修路花费为L。

输出

对于每组测试数据输出Yes或No（如果存在两种以上的最小花费方案则输出Yes,如果最小花费的方案只有一种，则输出No)

样例输入

<span style="color:#333333"><span style="color:black">2
3 3
1 2 1
2 3 2
3 1 3
4 4
1 2 2
2 3 2
3 4 2
4 1 2</span></span>

样例输出

<span style="color:#333333"><span style="color:black">No
Yes</span></span>

只要是找到跟该边长度相等并且顶点都在同一个连通分量上就一定存在第二种方案 

#include<stdio.h>
#include<algorithm>
using namespace std;
struct Node{
	int s,e,cost;
}edge[200005];
int pre[510]; 
int cmp(Node x,Node y){
	return x.cost<y.cost;
}
int find(int n){//寻找根节点 
	return n==pre[n]?n:find(pre[n]);
}
int main(){
	int T,V,E,x,y;
	scanf("%d",&T);
	while(T--){
		scanf("%d %d",&V,&E);
		for(int i=0;i<E;i++){
			scanf("%d %d %d",&x,&y,&edge[i].cost);
			edge[i].s=min(x,y);//这里得是s记录小值e记录大值（不是太懂，但最好还是分的清一些） 
			edge[i].e=max(x,y);
		}
		for(int i=1;i<=V;i++)
		pre[i]=i;//记录父结点 
		sort(edge,edge+E,cmp);
		int flag=0;
		for(int i=0;i<E;i++){
			int a=find(edge[i].s);
			int b=find(edge[i].e);
			if(a!=b){
				for(int j=i+1;j<E;j++){
					if(edge[j].cost!=edge[i].cost)
					break;
					else if(find(edge[j].s)==a&&find(edge[j].e)==b){
						flag=1;
						break;
					}
				}
				pre[a]=b;
			} 
		}
		if(flag)
		printf("Yes\n");
		else
		printf("No\n");
	}
	return 0;
}