NYOJ118 修路方案（次小生成树）

修路方案

时间限制： 3000 ms  |  内存限制： 65535 KB

难度： 5

描述

南将军率领着许多部队，它们分别驻扎在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)

样例输入

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

样例输出

No
Yes
/*
poj1679的中文版
初始化放到了输入V的后边，调试了很久 
加油！！！
Time:2014-10-17 11:28
*/
#include <cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define INF 1<<27
#define MAX 510
bool used[MAX][MAX],vis[MAX];
int map[MAX][MAX],path[MAX][MAX],pre[MAX];
int d[MAX];
int V,E;
int Prim(){
	int Mst=0;
	memset(vis,0,sizeof(vis));
	memset(used,0,sizeof(used));
	memset(d,0,sizeof(d)); 
	for(int i=1;i<=V;i++){
		d[i]=map[1][i];
		pre[i]=1;
	}
	vis[1]=true;d[1]=0;
	for(int i=1;i<=V;i++){
		int u=1;int minV=INF;
		for(int j=1;j<=V;j++){
			if(!vis[j]&&minV>d[j]){
				minV=d[j];
				u=j;
			}
		}
		vis[u]=true; 
		if(minV==INF)return Mst;
		Mst+=d[u];
	//	printf("Mst=%d\n",Mst);
		used[u][pre[u]]=used[pre[u]][u]=true;
		for(int j=1;j<=V;j++){
			if(vis[j]&&j!=u){
			path[u][j]=path[j][u]=max(path[pre[u]][j],d[u]);
			}
			if(!vis[j]&&d[j]>map[u][j]){
				d[j]=map[u][j];
				pre[j]=u;
			}
		}
	}
	return Mst;
}
int main(){
	int T;
	scanf("%d",&T);
	while(T--){
		int a,b,cost;
		scanf("%d%d",&V,&E);
		for(int i=0;i<=V;i++)
		for(int j=0;j<=V;j++){
			if(i==j)map[i][j]=0;	
			else {
				map[j][i]=map[i][j]=INF;
			path[i][j]=path[j][i]=0;
			used[j][i]=used[i][j]=0;
			}	
		}
		
		while(E--){
			scanf("%d%d%d",&a,&b,&cost);
			if(cost<map[a][b])
			map[a][b]=map[b][a]=cost;
		}
		int Mst=Prim();int ans=INF;
		for(int i=1;i<=V;i++){
			for(int j=1;j<=V;j++){
				if(i!=j&&!used[i][j])
				ans=min(Mst-path[i][j]+map[i][j],ans);
			}
		}
		//printf("Mst=%d ans=%d\n",Mst,ans);
		if(ans==Mst)printf("Yes\n");
		else printf("No\n");
	}
	return 0;
}