bzoj1048 [HAOI2007]分割矩阵

Description

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

Solution

最近怎么做了这么多蛤省河南省的题啊。
这道题做着真是发自内心的舒\(♂\)爽，几乎是最暴力的dp，不像有些题真心磨脑子。用\(s[i][j]\)记录二维前缀和，\(f[a][b][c][d][k]\)表示左上角为\((a, b)\)右下角为\((c, d)\)分成\(k\)块的方差总和。然后暴力记忆化搜索就好了

#include<cmath>
#include<queue>
#include<bitset>
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

#define rep(i, a, b) for (int i = a; i <= b; i++)
#define drp(i, a, b) for (int i = a; i >= b; i--)
#define fech(i, x) for(int i = 0; i < x.size(); i++)
#define N 15

inline int read() {
	int x = 0, flag = 1; char ch = getchar();
	while (ch > '9' || ch < '0') { if (ch == '-') flag = -1; ch = getchar(); }
	while (ch >= '0' && ch <= '9') { x = x * 10 + ch - '0'; ch = getchar(); }
	return x * flag;
}
inline void write(int x) { if (x >= 10) write(x / 10); putchar(x % 10 + '0'); }

double  f[N][N][N][N][N];
int s[N][N];
double ba;

double dfs(int a, int b, int c, int d, int k) {
	double tmp;
	if (f[a][b][c][d][k] != -1) return f[a][b][c][d][k];
	if (k == 1) {
		tmp = s[c][d] + s[a - 1][b - 1] - s[c][b - 1] - s[a - 1][d];
		return f[a][b][c][d][k] = (tmp - ba) * (tmp - ba);
	}
	int i, j; tmp = 1e9;
	rep(i, a, c - 1) rep(j, 1, k - 1) tmp = min(tmp, dfs(a, b, i, d, j) + dfs(i + 1, b, c, d, k - j));
	rep(i, b, d - 1) rep(j, 1, k - 1) tmp = min(tmp, dfs(a, b, c, i, j) + dfs(a, i + 1, c, d, k - j));
	return f[a][b][c][d][k] = tmp;
}
int main() {
	int n, m, k;
	scanf("%d%d%d", &n, &m, &k);
	rep(i, 1, n) rep(j, 1, m) s[i][j] = read() + s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1];
	rep(a, 0, n + 1) rep(b, 0, m + 1) rep(c, 0, n + 1) rep(d, 0, m + 1) rep(t, 0, k + 1)
		f[a][b][c][d][t] = -1;
	ba = (double)s[n][m] / k;
	printf("%.2f", sqrt(dfs(1, 1, n, m, k) / k));
	return 0;
}