记录编号 602735 评测结果 AAAAAAAAAA
题目名称 2034.[Tyvj]方块消除 最终得分 100
用户昵称 Gravatar李奇文 是否通过 通过
代码语言 C++ 运行时间 0.452 s
提交时间 2025-07-05 16:06:16 内存使用 27.17 MiB
显示代码纯文本
#include<bits/stdc++.h>
using namespace std;
const int N=205;
int n,a[N],f[N][N][N],cnt=1,ans;
vector<pair<int,int>>v;
int dfs(int l,int r,int k){
	if(l>r) return 0;
	if(f[l][r][k]!=-1) return f[l][r][k];
	int ans=dfs(l,r-1,0)+(v[r].second+k)*(v[r].second+k);
	for(int i=l;i<r;i++){
		if(v[i].first==v[r].first){
			ans=max(ans,dfs(l,i,v[r].second+k)+dfs(i+1,r-1,0));
		}
	}
	return f[l][r][k]=ans;
}

int main(){
	freopen("eliminate.in","r",stdin);
	freopen("eliminate.out","w",stdout);
	scanf("%d",&n);
	for(int i=0;i<n;i++){
		scanf("%d",&a[i]);
	}
	for(int i=0;i<n;i++){
		if(a[i+1]!=a[i]){
			v.push_back({a[i],cnt});
			cnt=1;
		}else{
			++cnt;
		}
	}
	for(int i=0;i<n;i++){
		for(int j=0;j<n;j++){
			for(int k=0;k<n;k++){
				f[i][j][k]=-1;
			}
		}
	}
	printf("%d\n",dfs(0,v.size()-1,0));
	return 0;
}