【BZOJ3754】Tree

Description

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

Wayne 拥有恰好N - 1 支修建队，每支队伍能且仅能修一条道路。当然，修建长度越大，修建的劳累度也越高，游戏设定是修建长度为C 的公路就会有C 的劳累度。当所有的队伍完工后，整个城市群必须连通，而这些修建队伍们会看看其他队伍的劳累情况，若劳累情况差异过大，可能就会引发骚动，不利于社会和谐发展。Wayne 对这个问题非常头疼，于是他想知道，这N - 1 支队伍劳累度的标准差最小能有多少。
标准差的定义：设有n的数$a_i$，它们的平均数是$\overline{a}$，那么标准差就是：$\sqrt{\dfrac{\sum_{i=0}^{n-1}(a_i-\overline{a})^2}{n}}$

注意：【BZOJ3754】并没有这道题。

Data Constraint

对于20% 的数据，M <= 20。

对于另外30% 的数据，Ci<= 10。

对于100% 的数据，N <= 100，M <= 2000，Ci <= 100。

Solution

对于20%数据，枚举每个选不选即可。

然后我们看无论是30%的还是100%的数据，它们的$C_i$都比较小。假设平均数为$\overline{a}$，那么答案就是：$\sqrt{\dfrac{\sum_{i=1}^{n-1}(a_i-\overline{a})^2}{n-1}}$

然后我们观察这个式子：显然要让$\sum_{i=1}^{n-1}(a_i-\overline{a})^2$最小，题目还要求选出来的所有边要使整个图联通。我们假设已经知道了$(a_i-\overline{a})^2$，那么求一个图的最小生成树可以完美解决这个问题。

从这个方面去想，我们自然而然想到枚举这个平均值，用$(a_i-\overline{a})^2$作为每个边的新边权来做最小生成树，但是这题要求精度太高，直接枚举会超时。

我们换个方面去想：首先所有原来边的边权都是整数，那么如果两个边权$a,b$平均数为$\overline{a'}$，那么无论我们在$[a,\overline{a'}]$（或者在$[\overline{a'},b]$ ）中取值，所做出的最小生成树是一样的（这句取自alan_cty原话），于是我们不必枚举如此高的精度，只要每次加0.25就行了。

但是，这样只保证了每次做出的最小生成树种类尽量是最小的，然而，我们求真正的答案，还需要把最小生成树上的边权取出来，重新计算标准差。

因为，这里$C_i≤100$，所以枚举的复杂度是$O(4maxC_i)$，最小生成树复杂度是$O(Nlog_2N)$的，所以总复杂的是$O(4maxC_iNlog_2N)$

Code

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define fd(i,j,k) for(int i=j;i>=k;i--)
#define N 101
#define M 2001
using namespace std;
struct node{
    int u,v,c;
    double z;
}b[M];
int f[N];
int n,m;
int find(int x)
{
    return !f[x]?x:f[x]=find(f[x]);
}
bool cmp(node x,node y)
{
    return x.z<y.z;
}
int main()
{
    cin>>n>>m;
    fo(i,1,m)
    {
        int u,v,c;
        scanf("%d %d %d",&u,&v,&c);
        b[i].c=c;
        b[i].u=u;
        b[i].v=v;
    }
    double ans=2147483647;
    for(double z=0.25;z<=100;z+=0.25)
    {
        fo(i,1,m) b[i].z=(b[i].c-z)*(b[i].c-z);
        sort(b+1,b+m+1,cmp);
        memset(f,0,sizeof(f));
        int cnt=0;
        int sqs=0,sum=0;
        fo(i,1,m)
        {
            int fx=find(b[i].u);
            int fy=find(b[i].v);
            if(fx!=fy)
            {
                f[fy]=fx;
                sum+=b[i].c;
                sqs+=b[i].c*b[i].c;
                cnt++;
                if(cnt==n-1) break;
            }
        }
        double ar=sum*1.0/(n-1);
        double p=sqrt((sqs-2*ar*sum+(n-1)*ar*ar)/(n-1));
        if(ans>p) ans=p;
    }
    printf("%.4lf",ans);
}