Main idea
有 n n n颗星星,编号为 1 1 1到 n n n,然后取出 k k k个数,要求这 k k k个数至少有 r r r个数是编号连续的。
Solution
设已经取了 i i i个的 r r r个连续,那在 n − k + 1 n-k+1 n−k+1种任意取 i i i块,然后对于剩下的 k − i ∗ r k-i*r k−i∗r可以在 n − i ∗ r n-i*r n−i∗r中任意取。
a n s = ∑ i = 1 ⌊ k r ⌋ ( − 1 ) i − 1 C n − k + 1 i C n − i ∗ r k − i ∗ r \large ans=\sum_{i=1}^{\left \lfloor \frac{k}{r} \right \rfloor}(-1)^{i-1}\mathbb{C}_{n-k+1}^{i}\mathbb{C}_{n-i*r}^{k-i*r} ans=i=1∑⌊rk⌋(−1)i−1Cn−k+1iCn−i∗rk−i∗r
Code
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
#define rep(i,x,y) for(int i=x;i<=y;i++)
using namespace std;
const int mod=1e9+7,M=1e7+5;
int n,k,r,inv[M+5],jc[M+5];
int ksm(int x,int y){
int ans=1;
for(;y;y>>=1,x=1ll*x*x%mod) if (y&1) ans=1ll*ans*x%mod;
return ans;
}
int C(int x,int y){
if (y>x) return 0;
return 1ll*jc[x]*inv[x-y]%mod*inv[y]%mod;
}
int main(){
scanf("%d%d%d",&n,&k,&r);
int z=k/r,t=1; long long ans=0;
jc[1]=jc[0]=inv[0]=1;
rep(i,2,n) jc[i]=1ll*jc[i-1]*i%mod;
inv[n]=ksm(jc[n],mod-2);
for(int i=n-1;i;i--) inv[i]=1ll*inv[i+1]*(i+1)%mod;
rep(i,1,z) ans=(ans+1ll*C(n-k+1,i)*C(n-i*r,k-i*r)%mod*t)%mod,t*=-1;
printf("%lld",(ans+mod)%mod);
return 0;
}