====== 可逆背包 ======
===== 问题简介 =====
考虑这样一个问题:有 $n$ 个物品,体积分别是 $w_1,w_2,\cdots,w_n$,背包容量为 $m$,共有 $n$ 次询问,第 $i$ 次询问要求不选物品 $i$ 时共有多少选法装满背包。
===== 思路分析 =====
如果对于每个询问依次求 01 背包问题,单次询问需要 $O(nm)$ 的时间,总复杂度 $O(n^2m)$。而退背包能让我们不需要对于每个询问都进行完整的 DP,而是用 $O(m)$ 时间去掉选择了物品 $i$ 的方案数,那么整个过程需要一次 $O(nm)$ 的DP和 $n$ 次 $O(m)$ 的退背包/加背包过程,整体复杂度就降到了 $O(nm)$。
===== 算法步骤 =====
1. 执行一次 01 背包算法过程
for(int j=m;j>=w[i];--j)
f[j]+=f[j-w[i]];
2. 对每个物品,将其看作是最后一个加入背包的物品,然后减掉它的方案数
memcpy(g,f,sizeof f);
for(int j=w[i];j<=m;++j)
g[j]-=g[j-w[i]];
===== 例题 =====
==== 例 1 ====
[[https://www.luogu.com.cn/problem/P4141|洛谷 P4141 消失之物]]
=== 题意 ===
有 $n$ 个物品,体积分别是 $w_1,w_2,\cdots,w_n$,现第 $i$ 个物品丢失,用剩下的 $n-1$ 个物品装满容积为 $x$ 的背包,有几种方法。把答案记为 $cnt(i,x)$,要得到所有 $i\in[1,n],x\in[1,m]$ 的 $cnt(i,x)$ 表格。输出 $n\times m$ 的矩阵,表示 $cnt(i,x)$ 的末位数字。
$1\le n,m\le 2000$。
=== 题解 ===
令 $f[j][0]$ 表示不算消失的物品组成容积为 $j$ 的方案数,即 01 背包问题,令 $f[j][1]$ 表示删掉某个物品后组成容积为 $j$ 的方案数。初始化是 $f[0][0]=f[0][1]=1$,即容积为 0 时方案数为 1
#include
using namespace std;
const int N=2e3+5;
int f[N][2],w[N];
int main(){
int n,m;
scanf("%d %d",&n,&m);
for(int i=1;i<=n;i++) scanf("%d",&w[i]);
f[0][0]=f[0][1]=1;
for(int i=1;i<=n;i++){
for(int j=m;j>=w[i];j--){
f[j][0]=(f[j][0]+f[j-w[i]][0])%10;
}
}
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
if(j-w[i]>=0) f[j][1]=(f[j][0]-f[j-w[i]][1]+10)%10;
else f[j][1]=f[j][0]%10;
printf("%d",f[j][1]);
}
puts("");
}
return 0;
}
==== 例 2 ====
[[https://codeforces.com/contest/1111/problem/D|CF 1111 D. Destroy the Colony]]
=== 题意 ===
给定一个长度为偶数的由大小写字母组成的字符串 $s$,一个“好”字符串可通过重排字母顺序使得所有相同字母的位置都在 $s$ 的同一半($[1,n/2] or [n/2+1,n]$)来生成。$q$ 个询问,给定一对 $(i,j)$,要求字母 $s[i],s[j]$ 也在 $s$ 的同一半的“好”字符串数量。
=== 题解 ===
注意到询问本质上最多只有 $52^2$ 种可能,则考虑先把所有询问算出来,最后 $O(1)$ 回答询问。
- 令 $k$ 为字母种数,统计各字母的出现次数 $c_1,c_2,\cdots,c_k$,若有 $i_1,i_2,\cdots,i_p$ 使得 $c_{i_1}+c_{i_2}+\cdots+c_{i_p}=\dfrac{n}{2}$,则把它们放到第一组,剩下的放到第二组,第一组的排列数为 $\dfrac{(\dfrac{n}{2})!}{c_{i_1}!c_{i_2}!\cdots c_{i_p}!}$,第二组排列数为 $\dfrac{(\dfrac{n}{2})!}{c_{j_1}!c_{j_2}!\cdots c_{j_s}!}$,总方法数为 $W=\dfrac{(\dfrac{n}{2})!(\dfrac{n}{2})!}{c_1!c_2!\cdots c_k!}$;
- 注意到 $c_i$ 求和为 $\dfrac{n}{2}$ 的组合数可用 01 背包来求,在没有 $(i,j)$ 限制下的“好”字符串数量为 $W*dp[n/2]*2$;
- 现考虑 $(i,j)$ 的限制,即字母 $s[i]$ 和 $s[j]$ 需在同一组中,等价于去掉字母 $s[i]$ 和 $s[j]$ 的 01 背包问题,即作了一次 01 背包问题后,再枚举 $i,j$ 做退背包过程,将去掉 $(i,j)$ 的 DP 值记作 $ans[i][j]$;
- 最后对每个询问直接 $O(1)$ 输出答案即可。
#include
using namespace std;
const int N=1e5+5,mod=1e9+7;
typedef long long ll;
ll fastpow(ll x,ll y){
ll ret=1;
for(;y;x=x*x%mod,y>>=1)
if(y&1) ret=ret*x%mod;
return ret;
}
int conv(int x){
if(x>='a') return x-'a'+26;
return x-'A';
}
ll dp[N],fac[N],inv[N],temp_dp[N];
ll ans[55][55];
ll w[55];
int main(){
string s;
cin>>s;
int n=s.length();
fac[0]=1;
w[conv(s[0])]++;
for(int i=1;i=0;i--) inv[i]=inv[i+1]*(i+1)%mod;
ll W=fac[n/2]*fac[n/2]%mod;
for(int i=0;i<52;i++){
if(w[i]) W=W*inv[w[i]]%mod;
}
dp[0]=1;
for(int i=0;i<52;i++){
if(w[i]){
for(int j=n;j>=w[i];j--){
dp[j]=(dp[j]+dp[j-w[i]])%mod;
}
}
}
for(int i=0;i<52;i++) ans[i][i]=dp[n/2];
for(int i=0;i<52;i++){
if(!w[i]) continue;
for(int k=0;k<=n;k++) temp_dp[k]=dp[k];
for(int k=w[i];k<=n;k++) temp_dp[k]=(temp_dp[k]-temp_dp[k-w[i]]+mod)%mod;
for(int j=i+1;j<52;j++){
if(!w[j]) continue;
for(int k=w[j];k<=n;k++) temp_dp[k]=(temp_dp[k]-temp_dp[k-w[j]]+mod)%mod;
ans[i][j]=2ll*temp_dp[n/2]%mod;
ans[j][i]=ans[i][j];
for(int k=n;k>=w[j];k--) temp_dp[k]=(temp_dp[k]+temp_dp[k-w[j]])%mod;
}
}
int q;
scanf("%d",&q);
while(q--){
int x,y;
scanf("%d%d",&x,&y);
int l=conv(s[x-1]);
int r=conv(s[y-1]);
printf("%lld\n",W*ans[l][r]%mod);
}
return 0;
}