图论_次小生成树_Kruskal算法_nyoj 118_修路方案

修路方案

时间限制： 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

来源

POJ题目改编

上传者

张云聪

2-次小生成树

思路：先求出最小生成树，得到最小生成树的权值min_w，再逐一去掉最小生成树的一条边进行遍历，看去掉这条边之后是否可以生成树，若可以生成树，再比较权值min2_w（次小生成树的权值）与min_w是否相等，相等则是次小生成树。

难点：怎么去边看能不能再生成一棵树，采用标记数组标记一下，这条边是否用过。

//Kruskal先求最小生成树，再求次小生成树

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
struct road{
	int x,y,w;
}rd[200001];
int n,m;
int par[505];
int vis[200001];
int cmp(road a,road b){
	return a.w<b.w;
} 
int find(int x)
{
	int i=x;
	while(!par[x]) return x;
	while(par[i]) i=par[i];
	int j=x,t;
	while(par[j])
	{
		t=par[j];
		par[j]=i;
		j=t;
	}
	return i;
}
int main()
{
	int t,nx,ny;
	bool f;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d%d",&n,&m);
		memset(par,0,sizeof(par));
		memset(rd,0,sizeof(rd));
		memset(vis,0,sizeof(vis));
		for(int i=0;i<m;i++)
		{
			scanf("%d%d%d",&rd[i].x,&rd[i].y,&rd[i].w);
		}
		sort(rd,rd+m,cmp);
		int ans=0;
		for(int i=0;i<m;i++)
		{
			nx=find(rd[i].x);
			ny=find(rd[i].y);
			if(nx!=ny){
				ans+=rd[i].w;
				par[nx]=ny;
				vis[i]=1;
			}
		}
	    int sum,res;
		bool flag=false;
		for(int i=0;i<m;i++)
		{
			if(vis[i]==1){
				vis[i]=2;
				res=0;
				sum=0;
				memset(par,0,sizeof(par));
				for(int j=0;j<m;j++)
				{
					if(vis[j]<2)
					{
						nx=find(rd[j].x);
						ny=find(rd[j].y);
						if(nx!=ny){
							res++;
							sum+=rd[j].w;
							par[nx]=ny;
			            }	
					}
				}
				f=true;
				if(res!=n-1)f=false;//判断是否能生成树 
				if(f)
				{//判断次小生成树的权值是否等于最小生成树 
					if(sum==ans){
						flag=true;
						break;
					}
				}
				vis[i]=1;//一定要在if里面还原 
			}
		}
		if(flag)
		printf("Yes\n");
		else
		printf("No\n"); 
	}
	return 0;
}