P3092 [USACO13NOV] No Change G
注意到 k <= 16,自然地想到状压dp。考虑 f S , i f_{S,i} fS,i 表示集合为 S,能否购买前 i 个,空间状态数 2 k × n 2^k\times n 2k×n 爆炸,由于 f S , / f_{S,/} fS,/ 有可分性(前一段为true 后一段为false),经典思路将判断可行性替换为极值,考虑 f S i f_{S_i} fSi 表示集合为 S,最远买到的 f S i f_{S_i} fSi,容易得到转移式 f S i ∣ t = max ( f S i ∣ t , f S i + d ) f_{S_i|t}=\max(f_{S_i|t},f_{S_i}+d) fSi∣t=max(fSi∣t,fSi+d),d 是硬币 t 能扩展的距离。复杂度 O ( 2 k × n k ) O(2^k\times n\ k) O(2k×n k),能过。
还能继续优化,注意到 d 与
f
S
i
f_{S_i}
fSi 无关,故 d 可以预处理出来。
具体的,
q
i
,
l
q_{\ i,l}
q i,l 表示硬币 i ,l 为起点时,到达的最远位置,求解拿个双指针尺取即可,复杂度
O
(
2
k
×
n
+
n
k
)
O(2^k\times n+n\ k)
O(2k×n+n k),洛谷跑总用时520ms。
被 -1 的情况卡了两发…
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5,M=(1<<17);
int n,k,a[20],c[N],q[20][N],f[M],ans=-1;
int main(){
scanf("%d%d",&k,&n);
for(int i=1;i<=k;i++) scanf("%d",a+i);
for(int i=1;i<=n;i++) scanf("%d",c+i);
for(int i=1;i<=k;i++){
int res=0;
for(int l=1,r=0;l<=n;res-=c[l++]){
while (r<n&&res+c[r+1]<=a[i]) res+=c[++r];
q[i][l]=r;
}
}
for(int S=0;S<(1<<k);S++){
int res=0;
for(int i=1;i<=k;i++){
if((S>>i-1)&1) continue;
res+=a[i];
f[S|(1<<i-1)]=max(f[S|(1<<i-1)],q[i][f[S]+1]);
}
if(f[S]==n) ans=max(ans,res);
}
printf("%d",ans);
}