算法提高 最小方差生成树 （枚举）

问题描述

给定带权无向图，求出一颗方差最小的生成树。

输入格式

输入多组测试数据。第一行为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 <cstdio>
#include<cstring>
#include<algorithm>

using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int maxm=1010;
const int maxn=110;
struct Edge{
    int u,v;
    int w;
    double f;
}edge[maxm];
/*
枚举平均值，然后计算
如果我们生成树的和不等于当前的sum的话，我们的值肯定是大于最优解的的，但是如果比当前的ans小的话，
肯定是ans比当前的ans还要小的,直接更新就好
这道题还是先去走，然后我们再去验证他的合理性，不一定非得每一步都得合理
*/

int n,m;
int pre[maxn];

int find_rt(int x){
    return pre[x]==x?x:pre[x]=find_rt(pre[x]);
}
void join(int u,int v,double& sum,double val){
    int ru=find_rt(u),rv=find_rt(v);
    if(ru!=rv){
        sum+=val;
        pre[ru]=rv;
    }
}

int cmp_f(Edge a,Edge b){
    return a.f<b.f;
}
double solve(double sum){
    double ave=sum/(n-1);
    for(int i=1;i<=m;i++){
      edge[i].f=(edge[i].w-ave)*(edge[i].w-ave);
    }
    sort(edge+1,edge+m+1,cmp_f);
    for(int i=1;i<=n;i++)pre[i]=i;
    double ans=0;
    for(int i=1;i<=m;i++){
        int u=edge[i].u,v=edge[i].v;
        join(u,v,ans,edge[i].f);
    }
  //  printf("sum:%.2f ans:%.2f\n",sum,ans);
    return ans;
}

int cmp_w(Edge a,Edge b){
    return a.w<b.w;
}
int main()
{
    int kase=1;
    while(scanf("%d %d",&n,&m)==2&&(n+m)){
        for(int i=1;i<=m;i++){
            scanf("%d %d %d",&edge[i].u,&edge[i].v,&edge[i].w);
        }
        sort(edge+1,edge+m+1,cmp_w);
        int min_sum=0,max_sum=0;
        for(int i=1;i<n;i++)min_sum+=edge[i].w;
        for(int i=m-n+2;i<=m;i++)max_sum+=edge[i].w;
        double ans=1e18;
        for(int sum=min_sum;sum<=max_sum;sum++){
            double tmp=solve((double)sum);
            ans=min(ans,tmp);
        }
        printf("Case %d: %.2f\n",kase++,ans*1.0/(n-1));
    }
    return 0;
}