P2112 鸿雁传书

#include<bits/stdc++.h>

#define MAXN 1005
#define MAXK 105

using namespace std;

int n,k;
double a[MAXN],f[MAXN][MAXK];//a[i]表示第i个单词的长度
double s[MAXN];//s是前缀和数组

int main()
{
    scanf("%d%d",&n,&k);
    double ave=0;//记录平均数
    for(int i=1; i<=n; i++)
    {
        char str[25];
        scanf("%s",str);//读入单词（全是x）
        a[i]=strlen(str);//记录单词长度
        ave+=a[i];//累加起来好算平均数
        s[i]=s[i-1]+a[i];//计算前缀和
    }
    ave/=k;//算出平均数


    for(int i=1; i<=n; i++)
        for(int j=2; j<=k; j++)
            f[i][j]=0x7fffffff;//除了第一列，其他地方都初始化成最大值
    //memset(f,0x7f,sizeof(f)); //二维数组 谨慎使用


    for(int i=1; i<=n; i++)
        f[i][1]=(s[i]-ave)*(s[i]-ave)/k;//分成一段的最小值直接计算出方差就行了

    for(int j=2; j<=k; j++) //分成一段不用处理了，直接从分成两段开始处理
        for(int i=j; i<=n; i++) //前面i个数最多也只能分成i段，所以如果j>i就不行
            for(int l=1; l<=i-1; l++)
                f[i][j]=min(f[i][j],f[l][j-1]+(s[i]-s[l]-ave)*(s[i]-s[l]-ave)/k);//转移方程

    printf("%.1lf",f[n][k]);//输出即可

    return 0;
}