记录编号	602649	评测结果	AAAAAAAAAA
题目名称	2034.[Tyvj]方块消除	最终得分	100
用户昵称	淮淮清子	是否通过	通过
代码语言	C++	运行时间	0.363 s
提交时间	2025-07-05 13:26:07	内存使用	27.53 MiB

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;

const int MAXN = 210;
int dp[MAXN][MAXN][MAXN];
vector<pair<int, int> > b;
int n, col[MAXN];

int dfs(int l, int r, int k){
    if(l > r) return 0;
    if(dp[l][r][k] != -1) return dp[l][r][k];
    int res = dfs(l, r - 1, 0) + (b[r].second + k) * (b[r].second + k);
    for(int m = l;m < r;m ++){
        if(b[m].first == b[r].first) res = max(res, dfs(l, m, b[r].second + k) + dfs(m + 1, r - 1, 0));
    }
    return dp[l][r][k] = res;
}

int main(){
	freopen("eliminate.in","r",stdin);
	freopen("eliminate.out","w",stdout);
    cin >> n;
    if(n == 0){
        cout << 0 << '\n';
        return 0;
    }
    for(int i = 0;i < n;i ++) cin >> col[i];
    int tmp = col[0];
    int cnt = 1;
    for(int i = 1;i < n;i ++){
        if(col[i] == tmp){
            cnt++;
        }
		else{
            b.emplace_back(tmp, cnt);
            tmp = col[i];
            cnt = 1;
        }
    }
    b.emplace_back(tmp, cnt);
    for(int i = 0;i < n;i ++){
    	for(int j = 0;j < n;j ++){
    		for(int k = 0;k < n;k ++){
    			dp[i][j][k] = -1;
			}
		}
	}
    cout << dfs(0, b.size() - 1, 0) << '\n';
    return 0;
}