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

来源:互联网 发布:淘宝卖家订单险多少钱 编辑:程序博客网 时间:2024/06/05 19:01

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

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

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

代码如下:

#include <cstdio>#include <algorithm>#include <cmath>#define N 120using 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]<sum[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;}
原创粉丝点击