修路方案
-
描述
-
南将军率领着许多部队,它们分别驻扎在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
-
第一行输入一个整数T(1<T<20),表示测试数据的组数
第一道次小生成树的题目,步骤是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;
}