FZU - 2282-Wand 错排公式

题意:

有n个人和n个椅子,每个人对应着一个椅子,现在要求至少有k个人坐在自己的位置上,然后其他人都不能做在自己的位置上

于是

ans = ∑C(n,i) * F(n-i) (k<=i<=n) 这里的F(n-i) 表示n - i 个错排的个数

错排公式:

当n个编号元素放在n个编号位置，元素编号与位置编号各不对应的方法数用D(n)表示，那么D(n-1)就表示n-1个编号元素放在n-1个编号位置，各不对应的方法数，其它类推.

第一步，把第n个元素放在一个位置，比如位置k，一共有n-1种方法;

第二步，放编号为k的元素，这时有两种情况:⑴把它放到位置n，那么，对于剩下的n-1个元素，由于第k个元素放到了位置n，剩下n-2个元素就有D(n-2)种方法;⑵第k个元素不把它放到位置n，这时，对于这n-1个元素，有D(n-1)种方法;

综上得到

D(n) = (n-1) [D(n-2) + D(n-1)]

特殊地，D(1) = 0, D(2) = 1.

代码:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn = 1e4+10;
const ll mod = 1e9+7;
ll d[maxn],n,k;
ll fac[maxn],inv[maxn];
ll power(ll a,ll b) {
    ll ans = 1;
    while(b) {
        if(b & 1) ans = ans * a % mod;
        a = a * a % mod;
        b >>= 1;
    }
    return ans;
}
void init() {
    d[0] = 1; d[1] = 0;d[2] = 1;
    for(int i=3;i<maxn;i++) {
        d[i] = ((i-1) * (d[i-1] + d[i-2])%mod)%mod;
    }
    fac[0] = fac[1] = 1;
    inv[0] = inv[1] = 1;
    for(int i=2;i<=maxn;i++){
        fac[i] = (i * fac[i-1]) % mod;
        inv[i] = power(fac[i],mod-2);
    }
}
ll C(ll n,ll k) {
    if(n == k || k == 0) return 1;
    ll ans = (fac[n] * inv[k] % mod * inv[n-k]) % mod;
    //printf("%lld\n",ans);
    return ans;
}
int main()
{
    init();
    int caset;scanf("%d",&caset);
    while(caset--) {
        scanf("%lld%lld",&n,&k);
        ll ans = 0;
        for(int i=k;i<=n;i++) {
            //printf("%lld %lld\n",C(n,k),d[n-i]);
            ans = (ans + C(n,i) * d[n-i] % mod) % mod;
        }
        printf("%lld\n",ans);
    }
    return 0;
}