第一次觉得自己组合数学还可以,也可能是题目太明显了
大致题意:有n个房间,每个房间里面都有1个人,然后发生了k次移动,问现在可能的情况有多少。
先思考,在k次移动后,可能会出现多少个房间的人数为0,很显然,房间人数为0的房间个数的取值范围是[0,min(k,n)]。
再思考,在有 i 个房间人数为0后,就需要把这 i 个人分配到 n-i 个房间中,所以问题就变成了:把 i 个人分配到 n-i 个房间中,那么这个就是组合数学的经典问题:把m个球放到n个盒子中,盒子不同,盒子可以为空,所以 公式:。
就是组合数学里面求组合 就是 C……的公式,不会打。。
所以,总体写法:枚举0的个数 i,同时计算从n个房间选 i 个房间的组合数c1,再去计算 上面的式子c2。最后结果是:
最后,代码:
#include<bits/stdc++.h>
using namespace std;
const int mod=1e9+7;
const int maxn=2e5+100;
long long int fac[maxn];
void init()
{
fac[0]=1;
for(int i=1;i<=200000;++i)
{
fac[i]=fac[i-1]*i%mod;
}
}
long long int quick_pow(int n,long long int base)
{
long long int ans=1;
while(n)
{
if(n&1)
{
ans*=base;
ans%=mod;
}
base*=base;
base%=mod;
n>>=1;
}
return ans;
}
long long int C(int a,int b)
{
long long int fz=fac[b];
long long int fm=fac[a]*fac[b-a]%mod;
return fz*quick_pow(mod-2,fm)%mod;
}
int main()
{
int n,k;
init();
cin>>n>>k;
k=min(n-1,k);
long long int ans=1;
if(k==1)
ans=0;
for(int i=1; i<=k; ++i)
{
ans+=C(i,n)*C(n-i-1,n-1)%mod;
//cout<<C(i,n)<<" "<<quick_pow(i,n-i)<<" "<<endl;
ans%=mod;
}
cout<<ans<<endl;
return 0;
}