算法提高 最小方差生成树
时间限制:1.0s 内存限制:256.0MB
问题描述
给定带权无向图,求出一颗方差最小的生成树。
输入格式
输入多组测试数据。第一行为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
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
Case 2: 0.00
数据规模与约定
1<=U,V<=N<=50,N-1<=M<=1000,0<=W<=50。数据不超过5组。
刚开始吧,很纳闷什么是最小方差生成树,其实就是边的权值-边的权值的平均值的最小和, 然后就是多次kruskal构建生成树,
注意最小值最大值的寻找和与方差的对应
#include <bits/stdc++.h>
using namespace std;
const int N = 1005;
const int MAX = 1.0e+308;
struct node
{
int s, t;
double val, w;
}edge[N];
int tmp[N];
int pre[N];
int n, m;
double ans;
bool cmp(node a, node b)
{
return a.val < b.val;
}
void init()
{
for(int i = 0; i <= n; i++) pre[i] = i;
return ;
}
int find(int x)
{
if(pre[x] == x) return x;
else
{
pre[x] = find(pre[x]);
return pre[x];
}
}
void kruskal(int sum)
{
init();
double ave = sum*1.0/(n-1); //每条边的均值
double sum_w = 0;
double sum_v = 0;
for(int i = 0; i < m; i++)
{
edge[i].val = (edge[i].w - ave)*(edge[i].w -ave); // 求每个边的方差
}
sort(edge, edge+m, cmp);
int num = 0;
for(int i = 0; i < m; i++) //遍历m条边
{
int tx = find(edge[i].s);
int ty = find(edge[i].t);
if(tx != ty)
{
pre[tx] = ty;
sum_w += edge[i].w;
sum_v += edge[i].val;
num++;
}
if(num >= n-1) break;
}
if((int)sum_w == sum)
ans = min(ans, sum_v/(n-1));
}
int main()
{
int cnt = 0;
while(scanf("%d %d", &n, &m))
{
if(!n && !m) break;
for(int i = 0; i < m; i++)
{
scanf("%d %d %lf", &edge[i].s, &edge[i].t, &edge[i].w);
tmp[i] = edge[i].w;
}
sort(tmp, tmp+m);
int maxx = 0, minx = 0;
for(int i = 0; i < n-1; i++)
{
minx += tmp[i];
}
for(int i = m-1; i > m-n; i--)
{
maxx += tmp[i];
}
ans = MAX;
for(int i = minx; i <= maxx; i++)
{
kruskal(i);
}
printf("Case %d: %.2lf\n", ++cnt, ans);
}
return 0;
}