nyoj118 修路方案之次小生成树

修路方案

描述

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

<pre name="code" class="html"><pre name="code" class="html"><span style="font-size:18px;">#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的路径中的最大比边(就是从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;
   //这个循环，如果已经找到了两个节点1 和 3 
   //因为，对于节点1 来说在整个图中，到其他点权值最小的边已经找到；而且那条边也已经被删除；
   //所以，在第一次循环找1 到其他点的权值最小的边，是没有用的；而且，x,y 也会被下次循环中 到节点3 权值最小的节点所覆盖；
   //所以，这循环总能找到 下一个最小生成树的节点(注意理解 "下一个节点" 是对于整个查找过程而言的 )
   //其实我觉得可以在这里调整i的值，这样就不用每次从第一个节点开始了，可是我换了之后，提交错了，不知道为什么？
        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;
                    }
                }
            }
        }

        /*上面的循环中找到，在生成最小生成树的过程中的下一个点;
        //而下面的这个循环,可以算出，从已经在的最小生成树中的节点 到 当前节点 的这条路径上的权值的最大值；
        //    比如ch2[5][1] = ch2[1][5] 记录的就是，从节点 1 到 5 这条路径上的边的权值的最大值;
        // 这道题虽然思想好理解，但是实现起来还真不简单啊；
        for(int i=0;i<s.size();i++)
        {
            //比如最小生成树已经有1 3 了这次又找到 节点5 这里就会比较 value_13 和value_35 
            // 如果最小生成树中已经有了1 3 5 这次又找到了6 这里就会用value_15(之前算出的) 和 value_56 比较
            //  这里的x,y可以这么理解，x表示在生成最小生成树的过程中，从x点出发找到的下一个最小树中的节点就是y;
            // 所以下面的这段代码就不难理解了,
            //  因为，从第一个节点到节点x这条路径上的最长边已经找到(s[i][x]),所以求从第一节点到节点y这条路径上的最大边
            //就直接比较,s[i][x]和ch1[x][y]的值就行啦；
            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;
	int i,j;
	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(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(i=1;i<=m;i++)
		{
			for(j=1;j<=m;j++)
			{
				if(ch1[i][j]!=-1)
				{
					int k=sum-ch2[i][j]+ch1[i][j];
					if(k==sum)
					{
						flat=1;break;
					}
				}
			}
			if(flat)break;
		}
		if(flat)
			printf("Yes\n");
		else 
			printf("No\n");
	}
}</span>