AtCoder - 5312 （组合数学）

题目链接
题意：有n个房间，每个房间有1个人，现在有k步可以让一个人从一个房间走到另一个房间，问经过这k步之后不同的排列方式是多少？
思路：首先你要明白一点经过k步以后一定是有k个房间的人数为0的，不然的话如果1走到2号房间，2号再走到1号房间这个是个毫无意义的操作，所以公式就是ans=∑C(n,i)*C(n-1,i),i的范围是【0，min（n-1，k）】，这个组合数的意义就是从n个人里选出i个房间，然后把这n个人分配的n-i个房间里，这个就是经典的隔板法了，所以n-i-1块板是C（n-1，n-i-1）=C（n-1,i)

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=2e5+5;
const int mod=1e9+7; 
ll f[maxn],inv[maxn];
ll quick(ll a,ll b){ 
    ll ans=1;  
    a=a%mod;
    while(b!=0){
        if(b&1) ans=(ans*a)%mod;
        b>>=1;
        a=(a*a)%mod;
    }
    return ans;
}
ll cal(int n,int k)
{
	return f[n]*inv[n-k]%mod*inv[k]%mod;
}
int main()
{
	int n,k;
	f[0]=inv[0]=1;
	for(int i=1;i<maxn;++i) f[i]=f[i-1]*i%mod,inv[i]=quick(f[i],mod-2); 
	scanf("%d %d",&n,&k);
	ll ans=0;
	for(int i=0;i<=min(n-1,k);++i) ans+=cal(n,i)*cal(n-1,i)%mod,ans%=mod;
	printf("%lld\n",ans);
}