算法提高 最小方差生成树

  算法提高 最小方差生成树  

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

#include<bits/stdc++.h>

using namespace std;

int a[55],rank[55];
int b[1005];
int n,m;
double ans;
struct node
{
	int u,v;
	double w,val;
}e[1005];

bool cmp(node a,node b)
{
	return a.w < b.w;
}

void init(int n)
{
	for (int i=1;i<=n;i++)
	{
		a[i] = i;
		rank[i] = 0;
	}
}

int find(int x)
{
	if (a[x]==x) return x;
	else return a[x] = find(a[x]);
}

void unite(int x,int y)//用rank进行优化，也可不用rank数组，直接相连
{
	x = find(x);
	y = find(y);
	if (x==y) return;
	if (rank[x]<rank[y])
		a[x] = y;
	else
	{
		a[y] = x;
		if (rank[x]==rank[y])
			rank[x]++;
	}
		
}

bool same(int x,int y)
{
	return find(x)==find(y);
}

void kruskal(int sum)
{
	int cnt = 0;
	init(n);
	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 (!same(u,v))
		{
			unite(u,v);
			all += e[i].val;
			f_all += e[i].w;
			cnt++;
		}
		if (cnt==n-1)
			break;
	}
	if (int(all)==sum)
		ans = min(ans,f_all);
}

int main()
{
	int k = 1;
	while(scanf("%d %d",&n,&m)&&n+m!=0)
	{
		int minv = 0,maxv = 0;
		ans = 100000000.0;
		for(int i=0;i<m;i++)
        {
            scanf("%d %d %lf", &e[i].u,&e[i].v,&e[i].val);
            b[i] = e[i].val;
        }
        sort(b,b + m);
        for(int i=0;i<n-1;i++)
            minv += b[i];
        for(int i=m-1;i>m-n;i--)//前n-1个最小的和 
            maxv += b[i];
        for (int i=minv;i<=maxv;i++)//后n-1个最大的和 
        	kruskal(i);
        ans = ans / (n-1);
        printf("Case %d: %.2lf\n",k++,ans);
	}
	return 0;
}