洛谷P2503 BZOJ2428 [HAOI2006]均分数据

题目描述：

已知N个正整数：A1、A2、……、An 。今要将它们分成M组，使得各组数据的数值和最平均，即各组的均方差最小。均方差公式如下：

输入输出格式
输入格式：
输入文件data.in包括：
第一行是两个整数，表示N,M的值（N是整数个数，M是要分成的组数）
第二行有N个整数，表示A1、A2、……、An。整数的范围是1–50。
（同一行的整数间用空格分开）
输出格式：
输出文件data.out包括一行，这一行只包含一个数，表示最小均方差的值(保留小数点后两位数字)。
输入输出样例
输入样例#1：
6 3
1 2 3 4 5 6
输出样例#1：
0.00
说明
样例解释：1和6、2和5、3和4分别为一组
【数据规模】
对于40%的数据，保证有K<=N <= 10，2<=K<=6
对于全部的数据，保证有K<=N <= 20，2<=K<=6

#include<iostream>#include<cstring>#include<cstdio>#include<algorithm>#include<cstdlib>using namespace std;const int MAXN = 25;double mid ,ans;int a[MAXN],p[MAXN],n,m;double Pow(double x) { return x*x; }double solve(){    int i,cnt=1,sum=0;    double t1,t2,ret=0;    for(i=1;i<=n&&cnt<m;i++){        if(sum+a[p[i]]>=mid){            t1=Pow(sum-mid);t2=Pow(sum-mid+a[p[i]]);//因为没说每组数据个数强制相等            //在此处做一下取舍            if(t1<t2) ret+=t1,sum=a[p[i]];            else ret+=t2,sum=0;            cnt++;        }        else sum+=a[p[i]];    }    while(i<=n) sum+=a[p[i++]];    ret+=Pow(sum-mid);    return ret;}int main(){    scanf("%d%d",&n,&m);    for(int i=1;i<=n;i++){        scanf("%d",&a[i]);        p[i]=i;        mid+=a[i];    }    mid/=m;ans=1e18; int T = 255498;    while(T--){        random_shuffle(p+1,p+n+1);        ans=min(ans,solve());    }    printf("%.2lf\n",sqrt(ans/m));    return 0;}