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>


 
    
 
   
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值