poj1390区间dp+记忆化搜索

题目大意：一排方块有颜色，每次只能消去相同颜色的块，分值为每次消去数量的平方，求最大和。

思路：区间dp，dp[i][j][k]表示从i到j块且j后面有k块与j不想领但同颜色。

则j可以和后面的合并也可以和前面的合并。具体看代码

#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <fstream>
#include <algorithm>
#include <cmath>
#include <queue>
#include <stack>
#include <vector>
#include <map>
#include <set>
#include <iomanip>

using namespace std;
//#pragma comment(linker, "/STACK:102400000,102400000")
#define maxn 210
#define MOD 1000000007
#define mem(a , b) memset(a , b , sizeof(a))
#define LL long long
#define INF 100000000

int n ;
int dp[maxn][maxn][maxn];
int arr[maxn];

int solve(int l , int r , int k)
{
    if(l > r ) return 0;
    if(dp[l][r][k]) return dp[l][r][k];
    dp[l][r][k] = solve(l , r-1 , 0) + (1+k) * (1+k);
    for(int i = r-1 ; i >= l ; i--)
    {
        if(arr[r] == arr[i])
        {
            dp[l][r][k] = max(dp[l][r][k] , solve(l , i , k + 1) + solve(i + 1 , r - 1, 0));
        }
    }
    return dp[l][r][k];
}

int main()
{
    int t ;
    int ncase = 1;
    scanf("%d" , &t);
    while(t--)
    {
        scanf("%d" , &n);
        for(int i = 1 ; i <= n ; i ++) scanf("%d" , &arr[i]);
        memset(dp , 0 , sizeof(dp));
        printf("Case %d: %d\n" , ncase ++ , solve(1 , n , 0));
    }
    return 0;
}