最小方差生成树 (Kruskal)

来源:互联网 发布:苹果随意定位软件 编辑:程序博客网 时间:2024/05/16 14:56

问题描述
给定带权无向图,求出一颗方差最小的生成树。
输入格式
输入多组测试数据。第一行为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组。


题解:首先说明LQB这题数据好像有错啊....这题有很多姿势。原题来自是BZOJ 3754 (VIP题).....

BZOJ那道是算标准差,这题是方差。姿势差不多一样。

对于这道题来说,可以这样做,因为求方差最小,那么就是要求每条边的(权重-边权和的平均值)^2的和最小。

那么我们枚举全部边权和的可能值,每次都跑一次Kruskal,跑的时候以(权重-边权和的平均值)^2为当前权重,然后如果我们先前枚举的的边权和等于跑完Kruskal后,生成树的权值。我们就更新答案就可以了。


代码:

#include<bits/stdc++.h>using namespace std;int n,m;double ans;int tmp[100100];int fa[100100]; struct edge{int x,y;double w,val;edge(int x=0,int y=0,double w=0,double val=0):x(x),y(y),w(w),val(val){}}e[100100];int cmp(edge a,edge b){return a.val<b.val;}int getfather(int x){return fa[x]==x ? x: fa[x]=getfather(fa[x]);}void kruskal(int sum){int cnt=n;double res=0;double tmp_ans=0;double average=sum * 1.0 /(n-1); //方差 * (n-1)for(int i=0;i<m;i++){e[i].val = (e[i].w-average) * (e[i].w-average);}for(int i=1;i<=n;i++)fa[i]=i;sort(e,e+m,cmp);for(int i=0;i<m;i++){int t1=getfather(e[i].x);int t2=getfather(e[i].y);if(t1!=t2){fa[t1]=t2;res += e[i].w;tmp_ans += e[i].val;cnt--;if(cnt==1)break;}}if((int)res==sum){ans=min(ans, tmp_ans);//cout<<"true"<<endl;} }int main(){//freopen("in.txt","r",stdin);int cas=1;while(~scanf("%d%d",&n,&m),n+m){int maxx=0,minn=0;ans=99999999999.0;for(int i=0;i<m;i++){cin>>e[i].x>>e[i].y>>e[i].w;tmp[i]=e[i].w;}sort(tmp,tmp+m);for(int i=0;i<n-1;i++){minn+=tmp[i];}for(int i=m-1;i>m-n;--i){maxx+=tmp[i];}for(int i=minn;i<=maxx;i++){ // kruskal(i);}ans=ans/(n-1);printf("Case %d: %.2f\n",cas++,ans);}return 0;}


1 0