poj 1390 Blocks

题目链接：http://poj.org/problem?id=1390

题目大意及思路：见黑书123页，很巧妙的一个dp,定义dp[i][j][k]为从第i到第j段方块且在最后加上k个与第j段方块颜色相同的方块的最大得分。其中k是为了dp而人为加上的东西，从而得到问题的解。

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<string>
#include<queue>
#include<algorithm>
#include<vector>
#include<stack>
#include<list>
#include<iostream>
#include<map>
using namespace std;
#define inf 0x3f3f3f3f
#define Max 210
inline int max(int a,int b)
{
	return a>b?a:b;
}
int min(int a,int b)
{
	return a<b?a:b;
}
struct node
{
    int next,val;
}e[2*Max];
int dp[Max][Max][Max];
int num[Max],co[Max];
int a[Max],p[Max],len[Max][Max];
int t,n;
int main()
{
    int i,j,k,l,g;
    int count=1;
    scanf("%d",&t);
    while(t--)
    {
        memset(num,0,sizeof(num));
        memset(len,0,sizeof(len));
        memset(p,-1,sizeof(p));
        int eid=0;
        j=-1;
        scanf("%d",&n);
        a[0]=-1;
        for(i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            if(a[i]!=a[i-1])
            {
                j++;
                num[j]++;
                co[j]=a[i];
            }
            else
            num[j]++;
        }
        for(i=j;i>=0;i--)
        {
          //  printf("%d %d\n",num[i],co[i]);
          e[eid].val=i;
          e[eid].next=p[co[i]];
          p[co[i]]=eid++;
          for(k=0;k<i;k++)
          len[k][co[i]]+=num[i];
        }
       // printf("a");
        int cnt=j;
        for(l=1;l<=cnt+1;l++)
        {

            for(i=0;i+l-1<=cnt;i++)
            {
                j=i+l-1;
                for(k=0;k<=len[j][co[j]];k++)
                {
                    if(i==j)
                    {
                        dp[i][j][k]=(num[j]+k)*(num[j]+k);
                        continue;
                    }
                    dp[i][j][k]=dp[i][j-1][0]+(num[j]+k)*(num[j]+k);
                    for(g=p[co[j]];g!=-1&&e[g].val<j;g=e[g].next)
                    {
                      //  printf("ak %d\n",e[g].val);
                      if(e[g].val>=i)
                        dp[i][j][k]=max(dp[i][j][k],dp[i][e[g].val][num[j]+k]+dp[e[g].val+1][j-1][0]);
                    }
                }
               // printf(" i %d j %d dp %d\n",i,j,dp[i][j][0]);
            }
        }
        //
        printf("Case %d: %d\n",count++,dp[0][cnt][0]);
    }
}