题目链接: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]);
}
}