【POJ 1390】Blocks

http://poj.org/problem?id=1390
黑书上的例题，感觉我这辈子是想不到这样的dp了QAQ
(f(i,j,k))表示将(i)到(j)合并，并且假设未来会有(k)个与(a_j)同色的方块与(j)相连的最大得分。
如果直接消去第(j)个区域和未来会接到(j)后面的(k)块,那么
(f(i,j,k)=f(i,j-1,0)+(b_j+k)^2)
如果(j)与之前一起合并，假设与(j)颜色相同的是区域(p)，那么
(f(i,j,k)=f(i,p,k+b_j)+f(p+1,j-1,0),(ileq p<j且a_p=a_j))
(f(i,j,k))为两者的最大值。
答案即是(f(1,n,0))
时间复杂度(O(n^4))。
(n^4)竟然能过！一直在想(n^3)QwQ（说得好像我(n^4)能想出来一样TwT）

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 203;
int in() {
	int k = 0; char c = getchar();
	for (; c < '0' || c > '9'; c = getchar());
	for (; c >= '0' && c <= '9'; c = getchar())
		k = k * 10 + c - 48;
	return k;
}

int T, n, f[N][N][N], a[N], len[N], b[N], m;

int sqr(int x) {return x * x;}

int dp(int l, int r, int ex) {
	if (r < l) return 0;
	if (f[l][r][ex] != -1) return f[l][r][ex];
	int ans = dp(l, r - 1, 0) + sqr(len[r] + ex);
	for (int i = l; i < r; ++i)
		if (a[i] == a[r])
			ans = max(ans, dp(l, i, len[r] + ex) + dp(i + 1, r - 1, 0));
	return f[l][r][ex] = ans;
}

int main() {
	T = in();
	for (int i = 1; i <= T; ++i) {
		printf("Case %d: ", i);
		memset(f, -1, sizeof(f));
		memset(len, 0, sizeof(len));
		m = in();
		for (int j = 1; j <= m; ++j)
			b[j] = in();
		a[n = 1] = b[1]; len[1] = 1;
		for (int j = 2; j <= m; ++j)
			if (b[j] != b[j - 1]) {
				a[++n] = b[j];
				len[n] = 1;
			} else
				++len[n];
		printf("%d
", dp(1, n, 0));
	}
	return 0;
}