AcWing 321. 棋盘分割

#include<bits/stdc++.h>
using namespace std;

const int N = 15, M = 9, INF = 1e9;
int g[M][M];
double f[M][M][M][M][N];//将(x1,y1)到(x2,y2)矩阵分割成k块所得最小均方差
int n,m=8;
double X;//平均值

double get(int x1,int y1,int x2,int y2)
{
    double sum=g[x2][y2]-g[x2][y1-1]-g[x1-1][y2]+g[x1-1][y1-1]-X;
    return (double)sum*sum;
}

double dp(int x1,int y1,int x2,int y2,int k)
{
    auto &v=f[x1][y1][x2][y2][k];
    if(v>=0)return v;
    if(k==1)return v=get(x1,y1,x2,y2);
    
    v=INF;
    //横切
    for(int i=x1;i<x2;i++)
    {
        v=min(v,dp(x1,y1,i,y2,k-1)+get(i+1,y1,x2,y2));
        v=min(v,dp(i+1,y1,x2,y2,k-1)+get(x1,y1,i,y2));
    }
    for(int i=y1;i<y2;i++)
    {
        v=min(v,dp(x1,y1,x2,i,k-1)+get(x1,i+1,x2,y2));
        v=min(v,dp(x1,i+1,x2,y2,k-1)+get(x1,y1,x2,i));
    }
    return v;

}

int main()
{
    cin>>n;
    for(int i=1;i<=m;i++)
    {
        for(int j=1;j<=m;j++)
        {
            scanf("%d",&g[i][j]);
            //计算(1,1)到(i,j)矩阵的前缀和
            g[i][j]+=g[i-1][j]+g[i][j-1]-g[i-1][j-1];
        }
    }
    X=(double)g[m][m]/n;
    memset(f,-1,sizeof f);
    
    printf("%.3lf",sqrt(dp(1,1,8,8,n)/n));
    return 0;
}