二项式反演笔记
二项式反演
若
\[f(n)=\sum^{n}_{i=0} g(i) C_{n}^{i}
\]
则
\[g(n)=\sum^{n}_{i=0}(-1)^{n-i}f(i)C_{n}^{i}
\]
证明:
代入\(g(i)\)得:
\[g(n)=\sum^{n}_{i=0}(-1)^{n-i}C_{n}^{i}\sum_{j=0}^{i}g(j)C_{i}^{j}
\]
改变枚举顺序可得:
\[g(n)=\sum^{n}_{i=0}(-1)^{n-i}C_{n}^{i}C_{i}^{j}\sum_{j=0}^{i}g(j)
\]
\[g(n)=\sum_{j=0}^{n}g(j)\sum^{n}_{i=0}(-1)^{n-i}C_{n}^{i}C_{i}^{j}
\]
我们可以通过暴力得到一个式子:
\[C_{n}^{i}C_{i}^{j}= \frac{n!}{i!(n-i)!}\times\frac{i!}{j!(i-j)!}=\frac{n!}{j!(n-j)!}\times\frac{(n-j)!}{(n-i)!(i-j)!}=C^{j}_{n}C^{n-i}_{n-j}
\]
把这个式子代入得:
\[g(n)=\sum_{j=0}^{n}g(j)\sum^{n}_{i=0}(-1)^{n-i}C^{j}_{n}C^{n-i}_{n-j}
\]
所以
\[g(n)=\sum_{j=0}^{n}g(j)C^{j}_{n}\sum^{n}_{i=0}(-1)^{n-i}C^{n-i}_{n-j}
\]
当\(i\ne n\)时,由二项式定理得:
\[\sum^{n}_{i=0}(-1)^{n-i}C^{n-i}_{n-j}=(-1+1)^{n-j}=0
\]
所以(此时\(i= n\)):
\[g(n)=g(n)C^{n}_{n}=g(n)
\]
证毕
例题
洛谷P4859 已经没有什么好害怕的了
由题意得,\(k=\frac{n+k}{2}\) (和差问题)
我们首先先将\(a,b\)从小到大排序
设\(f[i][j]\)表示前\(i\)个\(a\)中,选了\(j\)个\(a\)使得\(a>b\) 得方案数
令\(last[i]\)表示\(b\)中最后一个\(<a[i]\)的数的编号
无非就是选和不选,于是我们很容易得出状态转移方程:
\[f[i][j]=f[i-1][j]+(last[i]-j+1)*f[i-1][j-1]
\]
这里我们遇见了一个二项式反演题目中得一个常见套路:至少和恰好
令\(g[i]\)表示至少选了\(i\)个的方案数,\(ans[i]\)表示恰好选了\(i\)的方案数(就是我们要的答案)
在先前的dp中,我们还剩下一些没有选的数,它们可以任意排列
于是:
\[g[i]=f[n][i]*(n-i)!
\]
对于每个\(g[i]\),我们试图寻找它和所有比\(i\)大的\(j\)的\(ans[j]\)的关系
对于每个\(ans[j]\)中的每一种方案,我们可以从所有\(j\)个人中任意选择若干个人
所以我们得到:
\[g[i]=\sum_{j=i}^{n}C_{j}^{i}ans[j]
\]
由二项式反演得:
\[ans[j]=\sum_{j=i}^{n}(-1)^{j-i}C^{i}_{j}g[j]
\]
(代码被我Gu掉了)
#include<bits/stdc++.h>
#define int long long
#define MOD 1000000009
using namespace std;
const int maxn=2e3+5;
int n,k;
int a[maxn];
int b[maxn];
int dp[maxn][maxn];
int last[maxn];
int f[maxn];
int invf[maxn];
int g[maxn];
int ans[maxn];
inline int ksm(int x,int y){
if(!y) return 1;
if(y==1) return x%MOD;
int tmp=ksm(x,y/2)%MOD;
tmp=(1ll*tmp*tmp)%MOD;
if(y&1) return (tmp*x)%MOD;
else return tmp;
}
inline int calc(int x,int y){
return 1ll*f[x]*invf[y]%MOD*invf[x-y]%MOD;
}
signed main(){
scanf("%lld%lld",&n,&k);
for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
for(int i=1;i<=n;i++) scanf("%lld",&b[i]);
if((n+k)&1){
printf("0\n");
return 0;
}
sort(a+1,a+1+n);
sort(b+1,b+1+n);
f[0]=1;
invf[0]=1;
for(int i=1;i<=n;i++) f[i]=(1ll*f[i-1]*i)%MOD;
invf[n]=ksm(f[n],MOD-2);
for(int i=n-1;i>=1;i--) invf[i]=(1ll*invf[i+1]*(i+1))%MOD;
k=(k+n)/2;
for(int i=1;i<=n;i++) last[i]=lower_bound(b+1,b+1+n,a[i])-b-1;
dp[0][0]=1;
for(int i=1;i<=n;i++){
dp[i][0]=dp[i-1][0];
for(int j=1;j<=i;j++)
dp[i][j]=(1ll*dp[i][j]+(dp[i-1][j]+(1ll*dp[i-1][j-1]*max(0ll,last[i]-j+1))%MOD)%MOD)%MOD;
}
for(int i=1;i<=n;i++)
g[i]=(1ll*dp[n][i]*f[n-i])%MOD;
int ans=0;
for(int i=k;i<=n;i++)
if((i-k)&1) ans-=(1ll*calc(i,k)*g[i])%MOD,ans%=MOD;
else ans+=(1ll*calc(i,k)*g[i])%MOD,ans%=MOD;
while(ans<0) ans+=MOD;
printf("%lld\n",ans);
return 0;
}