考虑f[l][r]的最大值 , 我们可以强制将区间的某些数字弄到一起
那么f[l][r] 就等于
我们每次到l , r 就dfs 一遍 , 求出最大
#include<bits/stdc++.h>
#define N 55
using namespace std;
int n,c[N],v[N],f[N][N],next[N];
int Solve(int l,int r,int Sum,int Dp){
int ans = f[l+1][r] + Dp + Sum * Sum;
for(int i=next[l];i && i<=r;i=next[i]){
ans = max(ans , Solve(i,r,Sum+v[i],Dp+f[l+1][i-1]));
} return ans;
}
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%d",&c[i]);
for(int i=1;i<=n;i++) scanf("%d",&v[i]),f[i][i] = v[i]*v[i];
for(int i=1;i<=n;i++) for(int j=i+1;j<=n;j++) if(c[i]==c[j]){next[i] = j; break;}
for(int len=2;len<=n;len++) for(int l=1;l+len-1<=n;l++) f[l][l+len-1] = Solve(l,l+len-1,v[l],0);
printf("%d",f[1][n]); return 0;
}