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

第一道次小生成树的题目,步骤是prim算法求最小生成树的同时求出任意两点线路上的权值最大的边用max[][]表示,计算方法用到了动态规划,而且需要记录每个节点的父节点,用father[]表示,所以对于每一个节点c ,转移方程就是max[c][e] = max(max[e][father[c], max[father[c][c]),其中e表示除了c的父亲节点和c本身以外的其他节点,max[father[c]][c] 等于c与其父亲节点的边长,还有一个重要的问题就是prim算法里每次选完一个边,就要删除它,为以后的换边准备

做完上一步,任务就完成了一大半,这时候最小生成树的边都已经删完了,最小生成树的值也求出来了,记为ans,然后剩下的都是不选的边,这时候穷举所有的边,对于剩下的边(a,b) ,比较ans - max[a][b] + dis(a,b) ,dis(a,b)代表ab边的长度,取其中的最小值就是次小生成树,这个题题目说了是连通图,所以不用判断联通,poj1679没说是不是联通,所以要特殊判断

#include<stdio.h>
#include<string.h>
int map[510][510];
int max[510][510];
bool used[510];
int dis[510];
int father[510];
int v, e;

void prim(int *min_tree, int *second)
{
	memset(dis, 127, sizeof(dis));
	memset(max, 0, sizeof(max));
	memset(used, 0, sizeof(used));
	memset(father, 0, sizeof(father));
	int ans = 0;
	int i, j;
	dis[1] = 0;
	int min = 0x7fffffff, mark;
	for(j = 0; j < v; j++)
	{
		min = 0x7fffffff;
		for(i = 1; i <= v; i++)
		{
			if(!used[i])
			{
				if(dis[i] < min)
				{
					min = dis[i];
					mark = i;
				}
			}
		}
//		if(min > 2000000000)
//			return;
		used[mark] = 1;
		ans += min;		
		max[mark][father[mark]] = map[mark][father[mark]];
		max[father[mark]][mark] = map[mark][father[mark]];
		map[mark][father[mark]] = 0;
		map[father[mark]][mark] = 0;
		for(i = 1; i <= v; i++)
		{
			if(used[i] && i != mark)
			{
				max[mark][i] = max[father[mark]][i] > max[father[mark]][mark] ? max[father[mark]][i] : max[father[mark]][mark];
				max[i][mark] = max[mark][i];
			}
		}
		for(i = 1; i <= v; i++)
		{
			if(map[mark][i] && !used[i])
			{
				if(dis[i] > map[mark][i])
				{
					dis[i] = map[mark][i];
					father[i] = mark;
				}
			}
		}
	}

	*min_tree = ans;
	int second_tree = 0x7fffffff;
	for(i = 1; i <= v; i++)
	{
		for(j = 1; j <= v; j++)
		{
			if(map[i][j])
			{
				if(second_tree > ans - max[i][j] + map[i][j])
				{
					second_tree = ans - max[i][j] + map[i][j];
				}
			}
		}
	}
	*second = second_tree;
}
int main()
{
//	freopen("in.txt", "r", stdin);
	int t;
	scanf("%d", &t);
	while(t--)
	{
		memset(map, 0, sizeof(map));
		scanf("%d%d", &v, &e);
		int a, b, l;
		int i;
		for(i = 0; i < e; i++)
		{
			scanf("%d%d%d", &a, &b, &l);
			map[a][b] = l;
			map[b][a] = l;
		}
		a = 0;
		b = 1;
		prim(&a, &b);
//		printf("%d\n", b);
		if(a == b)
			printf("Yes\n");
		else
			printf("No\n");
	}
	return 0;
}