蓝桥杯 算法提高 最小方差生成树

 算法提高 最小方差生成树  

时间限制：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

样例输出

Case 1: 0.22
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;
}