poj 1390 Blocks

//poj 1390 解法很巧妙，见lrj黑书。。。
#include <iostream>
#include <algorithm>
using namespace std;

int n,m;
int color[205],len[205];
int dp[205][205][205];
int after[205];
int main()
{
    int t,a;
    scanf("%d",&t);
    for (int tm=1;tm<=t;tm++)
    {
          scanf("%d",&n);
    m=0;
    memset(len,0,sizeof(len));
          for (int i=1;i<=n;i++)
    {
    scanf("%d",&a);
    if (a==color[m]) len[m]++;
    else
    {
     color[++m]=a;
     len[m]=1;
    }
    }
    for (int i=m;i>=1;i--)
    {
     after[i]=0;
     for (int j=i+1;j<=m;j++)
      if (color[j]==color[i])
       after[i]+=len[j];
    }
      
    memset(dp,0,sizeof(dp));
    for (int l=1;l<=m;l++)
     for (int i=1;i<=m-l+1;i++)
     {
      int j=i+l-1;
      for (int k=0;k<=after[j];k++)
      {
       dp[i][j][k]=dp[i][j-1][0]+(len[j]+k)*(len[j]+k);
       for (int v=j-1;v>=i;v--)
       if (color[v]==color[j])
       {
        dp[i][j][k]=max(dp[i][j][k],dp[i][v][len[j]+k]+dp[v+1][j-1][0]);
       }
      }

     }
    printf("Case %d: %d/n",tm,dp[1][m][0]);

    

 }
   // system("pause");
    return 0;
}