BZOJ 1048: [HAOI2007]分割矩阵 记忆化搜索，二维前缀和

Description

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

　　第一行为3个整数，表示a,b,n的值。第二行至第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

解题思路：
1：首先要知道均方差是什么，它就是我们经常说的标准差。
2：记忆化搜索。预处理二维前缀和

代码如下:

//bzoj 1048
#include <bits/stdc++.h>
using namespace std;
#define rep(i, a, b) for(int i = a; i <= b; i++)
int n, m, K;
double ave;
int a[15][15], sum[15][15];
double dp[15][15][15][15][15];
double dfs(int a, int b, int c, int d, int k){
    double &res = dp[a][b][c][d][k];
    if(res != -1) return res;
    if(k == 0){
        res = sum[b][d] - sum[a - 1][d] - sum[b][c - 1] + sum[a - 1][c - 1];
        res = (res - ave) * (res - ave);
        return res;
    }
    res = 1e9;
    for(int i = a + 1; i <= b; i++){
        for(int j = 0; j < k; j++){
            res = min(res, dfs(a, i - 1, c, d, j) + dfs(i, b, c, d, k - j - 1));
        }
    }
    for(int i = c + 1;  i <= d; i++){
        for(int j = 0; j < k; j++){
            res = min(res, dfs(a, b, c, i - 1, j) + dfs(a, b, i, d, k - j - 1));
        }
    }
    return res;
}
int main(){
    scanf("%d%d%d", &n, &m, &K);
    rep(i, 0, 10){
        rep(j, 0, 10){
            rep(k, 0, 10){
                rep(l, 0, 10){
                    rep(t, 0, 10){
                        dp[i][j][k][l][t] = -1;
                    }
                }
            }
        }
    }
    rep(i, 1, n){
        rep(j, 1, m){
            scanf("%d", &a[i][j]);
        }
    }
    rep(i, 1, n){
        rep(j, 1, m){
            sum[i][j] = sum[i-1][j] + sum[i][j-1] - sum[i-1][j-1] + a[i][j];
        }
    }
    ave = (double)sum[n][m] / K;
    dfs(1, n, 1, m, K - 1);
    printf("%.2f\n", sqrt(dp[1][n][1][m][K-1] / K));
    return 0;
}