POJ 1390 Blocks 区间dp

http://poj.org/problem?id=1390

https://www.acwing.com/problem/content/324/

题意：

n个木块排成一列，每个木块都有一个颜色。

每次，你都可以点击一个木块，这样被点击的木块以及和它相邻并且同色的木块就会消除。

如果一次性消除了k个木块，那么就会得到k*k分。

给定你一个游戏初始状态，请你求出最高得分是多少

输入格式

第一行包含整数t，表示共有t组测试数据。

每组数据第一行包含整数n，表示共有n个木块。

第二行包含n个整数，表示n个木块的颜色。

代表木块颜色的整数范围是1~n。

输出格式

每组数据输出一个结果，每个结果占一行。

输出格式为“Case x: y”，其中x为数据组别编号，从1开始，y为结果。

数据范围

1≤n≤200

 

（orz。。。。）

首先对于初始状态下的区间必须同时合并，因此为了简化计算量，可以先将初始状态的连续区间缩点。

然后考虑[i,j]区间的最优解如何得到。

假如[1,5] 区间的颜色id分别为 1,2,3,4,2  (缩点后的区间），那么这个区间有可能由 区间[1,4]的最优解再直接和 [5]这个区间合并得到，也可能先削掉区间[3,4]，剩下的区间就是[1,2]+[5] 

但是实施起来会遇到一点问题

首先是如何表示区间[1,2]+[5]

其次是，随着[5]长度的不同，[1,5]区间的最优策略合并路线可能也不同，区间的长度影响着合并路线以及结果

因此引入第三维。。

dp[i][j][k]表示 第i点到第j点并且第j点之后有k个跟第j点一样的元素合并之后的最大值。。。

 

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#define ll long long 
#define ull unsigned long long
using namespace std;
const int INF = 0x3f3f3f3f;
const int maxn = 207;
//block_len:区间长度  block：区间id
int dp[maxn][maxn][maxn], block_len[maxn], block[maxn];
int solve(int l, int r,int extra) {
	if (dp[l][r][extra]) //记忆化搜索
		return dp[l][r][extra];
	if (l == r)
		return (block_len[l] + extra)*(block_len[l] + extra); //递归边界
	dp[l][r][extra] = solve(l, r - 1, 0) +(block_len[r] + extra)*(block_len[r] + extra);//边界直接合并
	for (int i = l; i < r; i++) { //先合并i+1~r 再合并l~i && k
		if (block[i] == block[r]) {
			dp[l][r][extra] = max(dp[l][r][extra], solve(i + 1, r - 1, 0) + solve(l, i, block_len[r] + extra));
		}
	}
	return dp[l][r][extra];
}
int main() {
	int t; cin >> t;
	for (int i = 1; i <= t; i++) {
		int n; cin >> n;
		memset(dp, 0, sizeof(dp));
		int tot = 0, inp;
		//压缩区间
		for (int i = 1; i <= n; i++) {
			scanf("%d", &inp);
			if (inp != block[tot]) {
				block[++tot] = inp;
				block_len[tot] = 1;
			}
			else
				block_len[tot]++;
		}
		printf("Case %d: %d\n", i, solve(1, tot, 0));
	}
	return 0;
}