bzoj2428 [HAOI2006]均分数据 模拟退火

题目大意:
把n个数分成m组,要求分的尽量平均,即方差最小,求这个方差。

题目分析:
http://blog.csdn.net/lych_cys/article/details/50843232
我们先随机一个答案,设置一个温度,然后把其中一个数从原来的分组中拿出来,随机放到另一个分组中,并计算答案,如果答案更优,就接受,否则就以一定概率(温度越高概率越大)接受这个答案,并且慢慢降温。

因为模拟退火很有可能接受一个更劣的答案,所以要多做几次来保证答案的正确性。

代码如下:

#include 
#include 
#include 
#define N 120
using namespace std;
inline double sqr(double x) { return x*x; }
int n,m,bl[N];
double a[N],sum[N],ave;
double annealing()
{
    sum[0]=1e100;
    for(int i=1;i<=m;i++) sum[i]=0;
    for(int i=1;i<=n;i++)
    {
        bl[i]=rand()%m+1;
        sum[bl[i]]+=a[i];
    }
    double T=1e4,last,ans=0;
    for(int i=1;i<=m;i++) ans+=sqr(sum[i]-ave);
    for(;T>0.1;T*=0.75)
    {
        last=ans;
        int x=rand()%n+1,y=0;
        if(T<500) y=rand()%m+1;
            for(int i=1;i<=m;i++) if(sum[i]y]) y=i;
        ans+=a[x]*(a[x]+sum[y]-sum[bl[x]])*2;
        if(ans>last && T<rand()%10000) ans=last;
        else { sum[bl[x]]-=a[x]; bl[x]=y; sum[y]+=a[x]; }
    }
    return ans;
}
int main()
{
    srand(20010312);
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%lf",&a[i]);
        ave+=a[i];
    }
    ave/=m;
    double ans=1e100;
    for(int i=1;i<=54321;i++) ans=min(ans,annealing());
    printf("%.2f",sqrt(ans/m));
    return 0;
}

你可能感兴趣的:(bzoj,模拟退火)