UVA - 10559 Blocks(DP)

#include <cstdio>
#include <cstring>
#include <iostream>
#include <map>
#include <vector>
#include <algorithm>
using namespace std;

const int maxn = 201;

int d[maxn][maxn][maxn],a[maxn],n;
bool vis[maxn][maxn][maxn];
int dp(int i,int j,int k){
if(vis[i][j][k]) return d[i][j][k];
vis[i][j][k] = true;

int& ans = d[i][j][k];
int p,q;
for(p = j;p>=i;p--) if(a[p]!=a[j]) {p++; break;}
if(p==i-1) p++;
if(p==i) return ans = (j - p + 1 + k)*(j - p + 1 + k);
ans = dp(i,p-1,0) + (j - p + 1 + k)*(j - p + 1 + k);
for(int q = p-1;q>=i;q--) if(a[q] == a[j] && a[q+1] != a[j]){
      ans =max(ans ,dp(i,q,k + j - p + 1)+dp(q+1,p-1,0));
}
return ans;
}
int main()
{
   int T,kase=1;
   scanf("%d",&T);
   while(T--){
       scanf("%d",&n);
       for(int i=1;i<=n;i++) scanf("%d",&a[i]);
       printf("Case %d: ",kase++);
       for(int i=0;i<=n;i++)
          for(int j=0;j<=n;j++)
             for(int k=0;k<=n;k++) vis[i][j][k] =false;
       printf("%d\n",dp(1,n,0));
   }
   return 0;
}