poj 1390 动态规划

思路：

黑书的例题

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<cstdio>
#define Maxn 210
using namespace std;
int dp[Maxn][Maxn][Maxn],col[Maxn],len[Maxn],pre[Maxn],aper[Maxn],after[Maxn],vi[Maxn];
int main()
{
    int t,n,i,j,x,k,Case=;
    int cnt=;
    scanf("%d",&t);
    while(t--){
        memset(dp,,sizeof(dp));
        memset(pre,,sizeof(pre));
        memset(len,,sizeof(len));
        memset(aper,,sizeof(aper));
        cnt=;
        scanf("%d",&n);
        for(i=;i<=n;i++){
            scanf("%d",&x);
            if(x!=col[cnt])
                col[++cnt]=x,len[cnt]=,pre[cnt]=aper[x],aper[x]=cnt;
            else
                len[cnt]++;
        }
        memset(vi,,sizeof(vi));
        for(i = cnt; i >= ; i--){
            if(vi[i]) continue;
            after[i] = ;
            int tmp = i;
            for(j = i-; j >= ; j--)
                if(col[j] == col[tmp]){
                    after[j] = after[tmp] + len[tmp];
                    tmp = j;
                    vi[j--] = true;
                }
        }
        int l;
        for(l=;l<=cnt;l++){
            for(i=;i+l<=cnt;i++){
                j=i+l;
                for(k=;k<=after[j];k++){
                    dp[i][j][k]=dp[i][j-][]+(len[j]+k)*(len[j]+k);
                    int p=pre[j];
                    while(p>=i){
                    dp[i][j][k]=max(dp[i][j][k], dp[i][p][k+len[j]]+dp[p+][j-][]);
                    p=pre[p];
                    }
                }
            }
        }
        printf("Case %d: %d\n",++Case,dp[][cnt][]);
    }
    return ;
}