WA #2 #3 #8 #10 求助
  • 板块P2135 方块消除
  • 楼主Lgx_Q
  • 当前回复0
  • 已保存回复0
  • 发布时间2022/8/30 08:15
  • 上次更新2023/10/27 13:08:16
查看原帖
WA #2 #3 #8 #10 求助
375953
Lgx_Q楼主2022/8/30 08:15
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxn=55;
int n,col[maxn],num[maxn],f[maxn][maxn],sn[maxn][maxn];
bool same[maxn][maxn];
signed main()
{
	scanf("%lld",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%lld",&col[i]);
	}
	for(int i=1;i<=n;i++)
	{
		scanf("%lld",&num[i]);
		same[i][i]=true;
		for(int j=i+1;j<=n;j++)
		{
			if(col[j]==col[j-1]) same[i][j]=same[i][j-1];
		}
		sn[i][col[i]]=num[i];
		for(int j=1;j<=n;j++)
		{
			sn[i][j]+=sn[i-1][j];
		}
	}
	for(int i=n;i>=1;i--)
	{
		for(int j=i;j<=n;j++)
		{
			if(same[i][j])
			{
				f[i][j]=(sn[j][col[i]]-sn[i-1][col[i]])*(sn[j][col[i]]-sn[i-1][col[i]]);
				continue;
			}
			for(int k=i;k<j;k++)
			{
				if(col[k]==col[j])
				{
					f[i][j]=max(f[i][j],f[i][k]+f[k+1][j-1]+2*num[j]*(sn[k][col[j]]-sn[i-1][col[j]])+num[j]*num[j]); //(a+b)^2->(a+b+c)^2
				}
				f[i][j]=max(f[i][j],f[i][k]+f[k+1][j]);
			}
		}
	}
	printf("%lld",f[1][n]);
	return 0;
}
2022/8/30 08:15
加载中...