问题重现
问题描述
给定带权无向图,求出一颗方差最小的生成树。
输入格式
输入多组测试数据。第一行为N,M,依次是点数和边数。接下来M行,每行三个整数U,V,W,代表连接U,V的边,和权值W。保证图连通。n=m=0标志着测试文件的结束。
输出格式
对于每组数据,输出最小方差,四舍五入到0.01。输出格式按照样例。
样例输入
4 5
1 2 1
2 3 2
3 4 2
4 1 1
2 4 3
4 6
1 2 1
2 3 2
3 4 3
4 1 1
2 4 3
1 3 3
0 0
样例输出
Case 1: 0.22
Case 2: 0.00
数据规模与约定
1<=U,V<=N<=50,N-1<=M<=1000,0<=W<=50。数据不超过5组。
AC代码
#include <cstdio>
#include <algorithm>
using namespace std;
double const MAX = 10000000000000.0;
int n, m, tmp[1005], fa[55];
double ans;
struct Edge
{
int u, v;
double w, val;
}e[1005];
bool cmp(Edge a, Edge b)
{
return a.w < b.w;
}
void UF_set(int n)
{
for(int i = 1; i <= n; i++)
fa[i] = i;
}
int Find(int x)
{
return x == fa[x] ? x : fa[x] = Find(fa[x]);
}
void Union(int a, int b)
{
int r1 = Find(a);
int r2 = Find(b);
if(r1 != r2)
fa[r2] = r1;
}
void Kruskal(int sum)
{
UF_set(n);
int cnt = 0;
double f_all = 0;
double all = 0;
double ave = sum * 1.0 / (n - 1);
for(int i = 0; i < m; i++)
e[i].w = (e[i].val - ave) * (e[i].val - ave);
sort(e, e + m, cmp);
for(int i = 0; i < m; i++)
{
int u = e[i].u;
int v = e[i].v;
if(Find(u) != Find(v))
{
Union(u, v);
f_all += e[i].w;
all += e[i].val;
cnt ++;
}
if(cnt == n - 1)
break;
}
if((int)all == sum)
ans = min(ans, f_all);
}
int main()
{
int ca = 1;
while(scanf("%d %d", &n, &m) != EOF && (m + n))
{
// if(n == 1 || n == 2)
// {
// printf("0.00\n");
// continue;
// }
int minv = 0;
int maxv = 0;
ans = MAX;
for(int i = 0; i < m; i++)
{
scanf("%d %d %lf", &e[i].u, &e[i].v, &e[i].val);
tmp[i] = e[i].val;
}
sort(tmp, tmp + m);
for(int i = 0; i < n - 1; i++)
minv += tmp[i];
for(int i = m - 1; i > m - n; i--)
maxv += tmp[i];
for(int i = minv; i <= maxv; i++)
Kruskal(i);
ans = ans / (n - 1);
printf("Case %d: %.2f\n", ca++, ans);
}
}
代码详解:
最小方差树的概念就是,在n个结点的图中,挑n-1个边,首先要使这个图连通,然后,这n-1条边的权值的方差要最小,方差的计算公式为:
可以看出,方差与边的平均值有关,但是有一个难点是,随着边的加入,边的平均值会动态地变化,很不利于解题。
这里有一个思路:在总的边的权和已知的情况下求解最小的方差,这是容易实现的,因为边权总和已知,然后结点数量(n)已知的话,边的数量也就知道了(n-1),这样边的平均值就知道了,这样各条边与均值的平方就是一个不动的常数了。此时我们再求最小生成树的过程中,边的权值就不是原来的权值了,而是其权值与平均值差值的平方,然后按照最小生成树的步骤求出最小的方差树。
使用克鲁斯卡尔算法求MST,使用到了并查集,代码来自https://www.cnblogs.com/asuml/p/6798307.html
可以看出,先按边的权值从大到小枚举,然后取其中n-1条边,这n-1条边的最大值就是从后往前取,最小值是从前往后取,然后最小生成树的代价就会在这两者之间,一一枚举,如果找到了有这样一个MST,那么求该情况下最小的方差,最后在所有可能的方差中取最小的。