[BZOJ 1048] [HAOI2007] 分割矩阵 【记忆化搜索】

题目链接：BZOJ - 1048

 

题目分析

感觉这种分割矩阵之类的题目很多都是这样子的。

方差中用到的平均数是可以直接算出来的，然后记忆化搜索 Solve(x, xx, y, yy, k) 表示横坐标范围 [x, xx]， 纵坐标范围 [y, yy] 的矩阵切成 k 块的最小 sigma((Vi - Ave)^2) 。

然后再递归将矩阵分得更小，直到 k 为 1 的时候直接返回相应的值。

 

代码

#include <iostream>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <cstdio>

using namespace std;

const int MaxN = 15 + 5, MaxT = 15 + 5;

int n, m, t, Num;
int Sum[MaxN][MaxN];

typedef double DB;

const DB INF = 999999999;

DB Ave;
DB f[MaxN][MaxN][MaxN][MaxN][MaxT];

DB Get(int x, int y, int xx, int yy) {
	return (DB)(Sum[xx][yy] - Sum[x - 1][yy] - Sum[xx][y - 1] + Sum[x - 1][y - 1]);
}

inline DB Sqr(DB x) {return x * x;} 
inline DB gmin(DB a, DB b) {return a < b ? a : b;}

DB Solve(int x, int xx, int y, int yy, int k) {
	if (f[x][xx][y][yy][k] != -1) return f[x][xx][y][yy][k];
	if (k == 1) return f[x][xx][y][yy][k] = Sqr(Get(x, y, xx, yy) - Ave);
	DB ret = INF;
	for (int i = x; i <= xx - 1; ++i)
		for (int j = 1; j <= k - 1; ++j)
			ret = gmin(ret, Solve(x, i, y, yy, j) + Solve(i + 1, xx, y, yy, k - j));
	for (int i = y; i <= yy - 1; ++i) 
		for (int j = 1; j <= k - 1; ++j) 
			ret = gmin(ret, Solve(x, xx, y, i, j) + Solve(x, xx, i + 1, yy, k - j));
	return f[x][xx][y][yy][k] = ret; 
}

int main() 
{
	scanf("%d%d%d", &n, &m, &t);
	for (int i = 1; i <= n; ++i)
		for (int j = i; j <= n; ++j)
			for (int p = 1; p <= m; ++p)
				for (int q = p; q <= m; ++q)
					for (int o = 1; o <= t; ++o) 
						f[i][j][p][q][o] = -1;
	memset(Sum, 0, sizeof(Sum));
	for (int i = 1; i <= n; ++i) {
		for (int j = 1; j <= m; ++j) {
			scanf("%d", &Num);
			Sum[i][j] = Sum[i][j - 1] + Sum[i - 1][j] - Sum[i - 1][j - 1] + Num;
		}
	}
	Ave = (DB)Sum[n][m] / (DB)t;
	Solve(1, n, 1, m, t);
	printf("%.2lf\n", sqrt(f[1][n][1][m][t] / t));
	return 0;
}