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

题目链接：http://acm.nyist.net/JudgeOnline/problem.php?pid=118

题意：在N个城市之间修路，在使任意两个城市都能相同的情况下修路的花费最少。问这种修路方案是否不止一种

解题思路：花费最少且任意两个城市能够相同，则说明要求最小生成树。而题目中问是否存在另外一种方案，达到

最小生成树的效果，所以可以采用次小生成树

常用的一种方法就是在求出最小生成树的基础上进行添加边

具体实现：先用prim算法求出最小生成树，并且统计任意一点到其他各点的路径上的最大边权。然后添加改生成树上没有的边（u,v），添加一条边后就会形成环，

然后删除该环中权值大二大的边（即除(u,v)之外的最大权值的边），然后再次统计此时的的费用，如果和最小生生成树的费用相同，则说明存在另外一种方案。

#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>
#include<cstdio>
using namespace std;
#define M 501
int ch1[M][M];   //保存原始边
int ch2[M][M];   //ch2[i][j]表示i到j的路径中的最大比边
vector<int> s;   //保存最小生成树的节点
int v[M];        //标记访问过的节点
int sum=0;
void prim(int m,int n)
{
	s.clear();s.push_back(m);v[m]=1;
	while(s.size()!=n)
	{
		int k=200000,x,y;
		for(int i=0;i<s.size();i++)
		{
			int r=s[i];
			for(int j=1;j<=n;j++)
			{
				if(!v[j]&&ch1[r][j]!=-1)
				{
					if(k>ch1[r][j])
					{
						k=ch1[r][j];x=r;y=j;
					}
				}
			}
		}
		for(int i=0;i<s.size();i++)
		{
			if(ch2[s[i]][x]<ch1[x][y])
			{ch2[y][s[i]]=ch2[s[i]][y]=ch1[x][y];}
			else {ch2[y][s[i]]=ch2[s[i]][y]=ch2[s[i]][x];}
		}
		s.push_back(y);sum+=k;v[y]=1;ch1[x][y]=-1;ch1[y][x]=-1;
	}
}
int main()
{
	int N;scanf("%d",&N);
	while(N--)
	{
		int m,n;scanf("%d%d",&m,&n);
		memset(ch1,-1,sizeof(ch1));memset(ch2,-1,sizeof(ch2));memset(v,0,sizeof(v));
		for(int i=0;i<n;i++)
		{
			int x,y,z;scanf("%d%d%d",&x,&y,&z);
			ch1[x][y]=z;ch1[y][x]=z;
			ch2[x][y]=z;ch2[y][x]=z;
		}
		prim(1,m);int flat=0;
		for(int i=1;i<=m;i++)
		{
			for(int j=1;j<=m;j++)
			{
				if(ch1[i][j]!=-1)
				{
					//cout<<i<<"   "<<j<<"   "<<ch1[i][j]<<endl;
					//cout<<ch2[i][j]<<endl;
					int k=sum-ch2[i][j]+ch1[i][j];
					if(k==sum)
					{
						flat=1;break;
					}
				}
			}
			if(flat)break;
		}
		if(flat)cout<<"Yes"<<endl;
		else cout<<"No"<<endl;
	}
}