UVA 10559 区间DP

/*题意就是一个一维的消灭星星的方块 问最大得分是多少*/
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<vector>
using namespace std;
typedef long long ll;
typedef pair<int,int> pp;
#define inf 0x3f3f3f3f
#define eps 1e-10
#define maxl 220
#define mem(i,j) memset(i,j,sizeof(i))
const int mod=1e9+7;
const double pi=acos(-1);
int a[maxl],n;
int d[maxl][maxl][maxl];//表示原序列的方块的右边再拼上k个颜色等于a[j]的方块所得到的新序列的最大分数

int dp(int i,int j,int k){
    if(i > j) return 0;
    int& ans = d[i][j][k];
    if(ans >= 0) return ans;
    int p = j;
    while(p >= i && a[p] == a[j]) p--;
    p++;
    ans=dp(i,p-1,0)+(j-p+k+1)*(j-p+k+1);//这里表示先把后面的这部分打掉 所以k=0
    for(int q=i;q<p;q++)
        if(a[q]==a[j]&&a[q]!=a[q+1])//找到那个分界点 就是中间有一段其他颜色的
            ans=max(ans,dp(i,q,j-p+k+1)+dp(q+1,p-1,0));//i~q 后面还有(j-p+k+1)个
    return ans;
}

int main()
{
    freopen("in.txt", "r", stdin);
    int t;
    cin>>t;
    for(int i=1;i<=t;i++){
        memset(d,-1,sizeof(d));
        cin>>n;
        for(int j=0;j<n;j++) cin>>a[j];
        printf("Case %d: %d\n",i,dp(0,n-1,0));
    }
}