洛谷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;
}