[HAOI2006][BZOJ2428] 均分数据

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<vector>
using namespace std;
int n,m,k,pos[21],a[21];
double ave,ans,mn=100000007,sum[21];
void solve()
{
    double t=10000; //初始温度
    ans=0;
    memset(sum,0,sizeof(sum));
    for (int i=1;i<=n;i++) pos[i]=rand()%m+1;
    for (int i=1;i<=n;i++) sum[pos[i]]+=a[i];
    for (int i=1;i<=m;i++) ans+=(sum[i]-ave)*(sum[i]-ave);
    while (t>0.1)
    {
        t*=0.91; //降温速率 0.9时WA。
//-----------------------------------------------------------------------
        int t1=rand()%n+1,x=pos[t1],y;  
        if (t>500) y=min_element(sum+1,sum+m+1)-sum; else y=rand()%m+1;   //随机选取两个位置 为什么>500取最小我也很不解，求解释。
//-----------------------------------------------------------------------
        if (x==y) continue;
        double pre=ans;
        ans-=(sum[x]-ave)*(sum[x]-ave);
        ans-=(sum[y]-ave)*(sum[y]-ave);
        sum[x]-=a[t1];
        sum[y]+=a[t1];
        ans+=(sum[x]-ave)*(sum[x]-ave);
        ans+=(sum[y]-ave)*(sum[y]-ave);                                   //换位置，更新ans，sum。
//-----------------------------------------------------------------------
        if (rand()%10000>t&&ans>pre)
        {
            sum[x]+=a[t1]; sum[y]-=a[t1];
            ans=pre;
        }
        else pos[t1]=y;
    }                                                                     //若不优，则返回原来的值，否则（小于概率或更优）更换位置。
//-----------------------------------------------------------------------
    if (ans<mn) mn=ans;
}
int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        ave+=a[i];
    }
    ave/=m;
    for (int i=1;i<=5000;i++) solve();
    printf("%.2lf",sqrt(mn/m));
    return 0;
}