考虑从小到大加入一个数,
加入
i
时会增加大于0小于
那么就相当于求 ∑ai=k 的方案数,其中 ai<i
这就是个很经典的背包了——BZOJ2431
但是这题不能用背包来做
考虑容斥。
朴素的容斥要枚举哪些超过限制,这样复杂度是指数级别的,但是很多是有重复的
令
fi,j
表示用
i
个数组成
gi=∑fi,j×(n+k−j−1n−1)
答案就是 ∑(−1)igi
fi,j 可以用旋转体积背包搞
#include <cstdio>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
const int N=200010,P=1e9+7;
int n,k,S;
ll f[510][N],inv[N],fac[N],a[N];
inline ll C(int x,int y){
return fac[x]*inv[y]%P*inv[x-y]%P;
}
int main(){
cin>>n>>k;
S=sqrt(k<<1)+3;
f[0][0]=1;
for(int i=1;i<=S;i++)
for(int j=i;j<=k;j++){
f[i][j]=(f[i][j-i]+f[i-1][j-i])%P;
if(j>n) (f[i][j]-=f[i-1][j-n-1])%=P;
}
fac[0]=inv[0]=inv[1]=1;
for(int i=1;i<=k+n;i++) fac[i]=fac[i-1]*i%P;
for(int i=2;i<=k+n;i++) inv[i]=1LL*(P-P/i)*inv[P%i]%P;
for(int i=1;i<=k+n;i++) (inv[i]*=inv[i-1])%=P;
ll ans=C(n+k-1,n-1);
for(int i=1;i<=S && i<=n;i++){
ll cur=0;
for(int j=1;j<=k;j++)
(cur+=f[i][j]*C(n+k-j-1,n-1))%=P;
if(i&1) cur=-cur;
(ans+=cur)%=P;
}
cout<<((ans+P)%P)<<endl;
return 0;
}