[BZOJ3754][JZOJ3410]【GDOI2014模拟】Tree

Description

Wayne 在玩儿一个很有趣的游戏。在游戏中，Wayne 建造了N 个城市，现在他想在这些城市间修一些公路，当然并不是任意两个城市间都能修，为了道路系统的美观，一共只有M 对城市间能修公路，即有若干三元组(Ui， Vi，Ci) 表示Ui 和Vi 间有一条长度为Ci 的双向道路。当然，游戏保证了，若所有道路都修建，那么任意两城市可以互相到达。

Wayne 拥有恰好N - 1 支修建队，每支队伍能且仅能修一条道路。当然，修建长度越大，修建的劳累度也越高，游戏设定是修建长度为C 的公路就会有C 的劳累度。当所有的队伍完工后，整个城市群必须连通，而这些修建队伍们会看看其他队伍的劳累情况，若劳累情况差异过大，可能就会引发骚动，不利于社会和谐发展。Wayne 对这个问题非常头疼，于是他想知道，这N - 1 支队伍劳累度的标准差最小能有多少。

Solution

最小方差生成树

我们可以枚举一个实数作为参考平均数，以每条边边权到这个数的差的绝对值排序做一遍最小生成树。

选出来的边，再求一次真实的平均数求标准差，更新答案。

Code

#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<iostream>
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fod(i,a,b) for(i=a;i>=b;i--)
using namespace std;
struct note
{
    int x,y;
    double z,v;
};
bool cmp(note x,note y)
{
    return x.z<y.z;
}
note rd[2005],a[2005];
int ft[105],n,m;
double p,s,ans,sum;
int getf(int k)
{
    if (ft[k]==0) return k;
    ft[k]=getf(ft[k]);
    return ft[k];
}
int main()
{
    cin>>n>>m;
    int i,j;
    fo(i,1,m)
    {
        scanf("%d%d%lf",&rd[i].x,&rd[i].y,&rd[i].z);
    }
    p=0;
    ans=1000000;
    while(p<=100)
    {
        p+=0.1;
        fo(i,1,m)
        {
            a[i].x=rd[i].x;
            a[i].y=rd[i].y;
            a[i].z=(rd[i].z-p)*(rd[i].z-p);
            a[i].v=rd[i].z;
        }
        sort(a+1,a+m+1,cmp);
        s=i=j=sum=0;
        double ave=0;
        memset(ft,0,sizeof(ft));
        while (j<n-1)
        {
            i++;
            int fx=getf(a[i].x),fy=getf(a[i].y);
            if (fx!=fy)
            {
                ft[fx]=fy;
                j++;
                sum+=a[i].v;
                s+=a[i].v*a[i].v;
            } 
        } 
        ave=sum/(n-1);
        s+=-2*ave*sum+(n-1)*ave*ave;
        ans=min(ans,sqrt(s/(n-1)));
    } 
    printf("%.4lf",ans); 
}