题目大意:一排方块有颜色,每次只能消去相同颜色的块,分值为每次消去数量的平方,求最大和。
思路:区间dp,dp[i][j][k]表示从i到j块且j后面有k块与j不想领但同颜色。
则j可以和后面的合并也可以和前面的合并。具体看代码
#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <fstream>
#include <algorithm>
#include <cmath>
#include <queue>
#include <stack>
#include <vector>
#include <map>
#include <set>
#include <iomanip>
using namespace std;
//#pragma comment(linker, "/STACK:102400000,102400000")
#define maxn 210
#define MOD 1000000007
#define mem(a , b) memset(a , b , sizeof(a))
#define LL long long
#define INF 100000000
int n ;
int dp[maxn][maxn][maxn];
int arr[maxn];
int solve(int l , int r , int k)
{
if(l > r ) return 0;
if(dp[l][r][k]) return dp[l][r][k];
dp[l][r][k] = solve(l , r-1 , 0) + (1+k) * (1+k);
for(int i = r-1 ; i >= l ; i--)
{
if(arr[r] == arr[i])
{
dp[l][r][k] = max(dp[l][r][k] , solve(l , i , k + 1) + solve(i + 1 , r - 1, 0));
}
}
return dp[l][r][k];
}
int main()
{
int t ;
int ncase = 1;
scanf("%d" , &t);
while(t--)
{
scanf("%d" , &n);
for(int i = 1 ; i <= n ; i ++) scanf("%d" , &arr[i]);
memset(dp , 0 , sizeof(dp));
printf("Case %d: %d\n" , ncase ++ , solve(1 , n , 0));
}
return 0;
}