由於每次合併可以刻畫為向外延伸,那麼考慮區間 \(\text{dp}\),設 \(dp_{l,r,m,c}\) 表示考慮了 \([l,r]\) 且剩下了一個質量為 \(m \in [0,K)\) 顏色為 \(c\) 的史萊姆的答案。
狀態過大且轉移方程不便於最佳化而考慮最佳化狀態,由於對於一個極短的需要合併成一個史萊姆的區間 \([p,q]\),最終的顏色只能是 \(c_p\) 或 \(c_q\),中間的顏色不可能延展出去否則會與極短不符。
所以設 \(f_{l,r}\) 表示處理掉 \([l,r]\) 的所有史萊姆的答案,\(fl_{l,r,p}\) 表示剩下一個質量為 \(p\) 且顏色為 \(c_l\) 的史萊姆的答案,\(fr_{l,r,p}\) 則是顏色為 \(c_r\) 的。
至於 \(p_i-p_{i-1}< w\) 則表示要合併的史萊姆並不會手動提升它的質量。
先考慮轉移 \(fl_{l,r,p}\),首先可以直接刪除 \((l,r]\),即 \(fl_{l,r,m_l}=f_{l+1,r}\),否則可以先保留 \(l\),再刪除中間一段區間,再把剩下的 \(fl\) 和 \(l\) 這個史萊姆合併,即 \(fl_{l,r,p}=\min_{l< k \le r,c_l=c_k} f_{l+1,k-1}+fl_{k,r,p-m_l}\),\(fr_{l,r,p}\) 同理可得。
至於 \(f_{l,r}\) 的轉移分為是否是一個極短合併區間,如果需要合併首先需要滿足 \(c_l=c_r\),那麼 \(f_{l,r}=\min_{l\le k< r,0< i< K,0< j< K}fl_{l,k,i}+fr_{k+1,r,j}+p_{i+j}\),如果不合並而直接合並兩邊的 \(f\),那麼 \(f_{l,r}=\min_{l\le k< r,c_k\neq c_{r+1} \vee c_{k+1}\neq c_{l-1}}\),\(c\) 的限制是因為不滿足就會自動與 \([l,r]\) 外的史萊姆合併。
可能有寫錯的地方記得來打我,可能有人要問 \(fl,fr,f\) 是不是還應該有其他關於 \(c\) 的限制啊,應該是存在一種順序滿足條件吧,其實我也不大知道,加上就過不了。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int n,s,w,c[151],m[151],p[21];
ll f[151][151],fl[151][151][21],fr[151][151][21];
void upd(ll &x,ll y){x=max(x,y);}
int main(){
memset(f,-0x3f,sizeof(f)),memset(fl,-0x3f,sizeof(fl)),memset(fr,-0x3f,sizeof(fr));
scanf("%d%d%d",&n,&s,&w);
for(int i=1;i<=n;i++) scanf("%d",&c[i]);
for(int i=1;i<=n;i++) scanf("%d",&m[i]);
for(int i=s;i<=2*s-2;i++) scanf("%d",&p[i]);
for(int i=1;i<s;i++) p[i]=p[s]-(s-i)*w;
for(int i=1;i<=n;i++) f[i][i]=p[m[i]],fl[i][i][m[i]]=fr[i][i][m[i]]=f[i][i-1]=0;
for(int t=2;t<=n;t++){
for(int l=1;l+t-1<=n;l++){
int r=l+t-1;
upd(fl[l][r][m[l]],f[l+1][r]);
for(int k=l+1;k<=r;k++) if(c[l]==c[k]) for(int i=1;i<s-m[l];i++) upd(fl[l][r][i+m[l]],f[l+1][k-1]+fl[k][r][i]);
upd(fr[l][r][m[r]],f[l][r-1]);
for(int k=l;k<r;k++) if(c[r]==c[k]) for(int i=1;i<s-m[r];i++) upd(fr[l][r][i+m[r]],f[k+1][r-1]+fr[l][k][i]);
for(int k=l;k<r;k++) if(c[k]!=c[r+1]||c[k+1]!=c[l-1]) upd(f[l][r],f[l][k]+f[k+1][r]);
if(c[l]==c[r]) for(int k=l;k<r;k++) for(int i=1;i<s;i++) for(int j=1;j<s;j++) upd(f[l][r],fl[l][k][i]+fr[k+1][r][j]+p[i+j]);
}
}
printf("%lld",f[1][n]);
return 0;
}