【BZOJ4517】【递推】【错排】排列计数 题解

4517: [Sdoi2016]排列计数

Description

求有多少种长度为 n 的序列 A，满足以下条件：
1 ~ n 这 n 个数在序列中各出现了一次
若第 i 个数 A[i] 的值为 i，则称 i 是稳定的。序列恰好有 m 个数是稳定的
满足条件的序列可能很多，序列数对 10^9+7 取模。
Input

第一行一个数 T，表示有 T 组数据。
接下来 T 行，每行两个整数 n、m。
T=500000，n≤1000000，m≤1000000
Output

输出 T 行，每行一个数，表示求出的序列数

Sample Input

5

1 0

1 1

5 2

100 50

10000 5000
Sample Output

0

1

20

578028887

60695423
HINT

Source

鸣谢Menci上传

序列里任意m个数稳定都是等价的，考虑不稳定的那几个就是裸错排
所以错排*组合数即可 f[n-m]*C(m,n)
设f[i]表示i个人的错排方案数
f[0]=1，f[1]=0
f[i+1]=(f[i]+f[i-1])*i
考虑第i+1个人怎么换，可以和前i个人里的一个互换，这是f[i]*i。或者拿第j个人的，并且第j个人不拿i的，在不考虑i的情况下第j个人不拿i与第j个人不拿j是等价的，所以就是f[i-1]*i。

#include <iostream>
#include <cstring>
#include <cstdio>
#define ll long long

const int maxn = 1000010;
const int MOD = 1000000007;
ll fac[maxn],ine[maxn],f[maxn];
int n,m,t;

template <class T> inline void read(T &x) {
    int flag = 1; x = 0;
    register char ch = getchar();
    while(ch <  '0' || ch >  '9') { if(ch == '-')  flag = -1; ch = getchar(); }
    while(ch >= '0' && ch <= '9') { x = (x<<1)+(x<<3)+ch-'0'; ch = getchar(); }
    x *= flag;
}

inline void init() {
    fac[0] = ine[0] = ine[1] = 1;  
    for(int i = 1; i < maxn; i++) fac[i] = fac[i-1]*i%MOD;
    for(int i = 2; i < maxn; i++) ine[i] = (MOD-MOD/i)*ine[MOD%i]%MOD;
    for(int i = 2; i < maxn; i++) ine[i] = ine[i-1]*ine[i]%MOD;
    f[0] = 1; f[1] = 0; f[2] = 1;  
    for(int i = 3; i <= 1000000; i++) f[i] = (f[i-1]+f[i-2])*(i-1)%MOD;
}

inline ll solve(int n, int m) { return fac[n]*ine[m]%MOD*ine[n-m]%MOD*f[n-m]%MOD; }

int main() {
    init();
    read(t);
    while(t--) read(n), read(m), printf("%lld\n",solve(n, m));
    return 0;
}