传送门
https://www.luogu.org/problemnew/show/P2217
思路
- 本质上就是dp,也可以说是记忆化搜索
- 设计状态f[a][b][c][d][t] 表示以(a,c)为起点(b,d)为终点的矩阵分t次的最小
(xi-ave)^2 - 转移就是枚举当前矩阵内可以分割的线,枚举分给左后两边分割的次数
代码
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
const double inf=1e9;
int m,n;
int k;
const int N=20;
int map[N][N];
double f[N][N][N][N][N];
int s[N][N];
double ave=0;
double dfs(int a,int b,int c,int d,int t)
{
double &tmp=f[a][b][c][d][t];
if(tmp!=-1) return tmp;
if(t==0)
{
tmp=s[b][d]-s[a-1][d]-s[b][c-1]+s[a-1][c-1];
tmp=(tmp-ave)*(tmp-ave);
return tmp;
}
tmp=inf;
for(int i=a+1;i<=b;i++)
for(int j=0;j<t;j++)
tmp=min(tmp,dfs(a,i-1,c,d,j)+dfs(i,b,c,d,t-j-1));
for(int i=c+1;i<=d;i++)
for(int j=0;j<t;j++)
tmp=min(tmp,dfs(a,b,c,i-1,j)+dfs(a,b,i,d,t-j-1));
return tmp;
}
int main()
{
cin>>n>>m>>k;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
cin>>map[i][j];
for(int a=0;a<=10;a++)
for(int b=0;b<=10;b++)
for(int c=0;c<=10;c++)
for(int d=0;d<=10;d++)
for(int t=0;t<=10;t++)
f[a][b][c][d][t]=-1;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]+map[i][j];
ave=(double)s[n][m]/k;
dfs(1,n,1,m,k-1);
printf("%.2lf",sqrt(f[1][n][1][m][k-1]/k));
return 0;
}