bzoj1048: [HAOI2007]分割矩阵

三十年来ac第一题。。。

1048: [HAOI2007]分割矩阵

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 818  Solved: 593
[Submit][Status][Discuss]

Description

　　将一个a*b的数字矩阵进行如下分割：将原矩阵沿某一条直线分割成两个矩阵，再将生成的两个矩阵继续如此
分割（当然也可以只分割其中的一个），这样分割了（n-1)次后，原矩阵被分割成了n个矩阵。（每次分割都只能
沿着数字间的缝隙进行）原矩阵中每一位置上有一个分值，一个矩阵的总分为其所含各位置上分值之和。现在需要
把矩阵按上述规则分割成n个矩阵，并使各矩阵总分的均方差最小。请编程对给出的矩阵及n，求出均方差的最小值
。

Input

　　第一行为3个整数，表示a,b,n(1<a,b<=10,1<n<=10）的值。
第二行至第n+1行每行为b个小于100的非负整数，表示矩阵中相应位置上的分值。每行相邻两数之间用一个空
格分开。

Output

　　仅一个数，为均方差的最小值（四舍五入精确到小数点后2位）

Sample Input

5 4 4
2 3 4 6
5 7 5 1
10 4 0 5
2 0 2 3
4 1 1 1

Sample Output

0.50

感觉好简单，一次就ac了

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
double a[11][11][11][11][11];
bool  bk[11][11][11][11][11];
int b[11][11],he[11][11];
int n,m,t;
double pingjun;
 
double pan(int x1,int y1,int x2,int y2,int ss)
{
    if (bk[x1][y1][x2][y2][ss])return a[x1][y1][x2][y2][ss];
    if (ss==1){
        double tsum=he[x2][y2]-he[x1-1][y2]-he[x2][y1-1]+he[x1-1][y1-1];
        a[x1][y1][x2][y2][ss]=(tsum-pingjun)*(tsum-pingjun);
        bk[x1][y1][x2][y2][ss]=true;
        //printf("%d %d %d %d %d:%lf\n",x1,y1,x2,y2,ss,a[x1][y1][x2][y2][ss]);
        return a[x1][y1][x2][y2][ss];
    }
    else
    {
    a[x1][y1][x2][y2][ss]=99999999.0;
    for (int i=x1;i<x2;i++){
    for (int x=1;x<ss;x++){
    a[x1][y1][x2][y2][ss]=
    min(a[x1][y1][x2][y2][ss],
pan(x1,y1,i,y2,x)+pan(i+1,y1,x2,y2,ss-x));
}
}
for (int i=y1;i<y2;i++){
    for (int x=1;x<ss;x++){
    a[x1][y1][x2][y2][ss]=
    min(a[x1][y1][x2][y2][ss],
pan(x1,y1,x2,i,x)+pan(x1,i+1,x2,y2,ss-x));
}
}
bk[x1][y1][x2][y2][ss]=true;
//printf("%d %d %d %d %d:%lf\n",x1,y1,x2,y2,ss,a[x1][y1][x2][y2][ss]);
        return a[x1][y1][x2][y2][ss];
    }
}
int main()
{
    int summ=0;
    memset(he,0,sizeof(he));
    scanf("%d %d %d",&n,&m,&t);
    for (int i=1;i<=n;i++)
        for (int j=1;j<=m;j++){
            scanf("%d",&b[i][j]);
            summ+=b[i][j];
            he[i][j]=he[i-1][j]+he[i][j-1]-he[i-1][j-1]+b[i][j];
        }
    pingjun=double(summ)/double(t);
    //printf("pj:%lf\n",pingjun);
    memset(a,-1,sizeof(a));memset(bk,false,sizeof(bk));
    printf("%0.2lf\n",sqrt(pan(1,1,n,m,t)/double(t)));
    
    return 0;
}