P4071 [SDOI2016]排列计数

题目描述

求有多少种长度为 n 的序列 A，满足以下条件：

1 ~ n 这 n 个数在序列中各出现了一次

若第 i 个数 A[i] 的值为 i，则称 i 是稳定的。序列恰好有 m 个数是稳定的

满足条件的序列可能很多，序列数对 10⁹+7 取模。

输入输出格式

输入格式：

第一行一个数 T，表示有 T 组数据。

接下来 T 行，每行两个整数 n、m。

输出格式：

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

输入输出样例

输入样例#1：

输出样例#1：

说明

测试点 1 ~ 3： T=1000，n≤8，m≤8；

测试点 4 ~ 6： T=1000，n≤12，m≤12；

测试点 7 ~ 9： T=1000，n≤100，m≤100；

测试点 10 ~ 12：T=1000，n≤1000，m≤1000；

测试点 13 ~ 14：T=500000，n≤1000，m≤1000；

测试点 15 ~ 20：T=500000，n≤1000000，m≤1000000。

Solution：

　　本题组合数学+错排公式+线推逆元。

　　组合数学和逆元就不说了，介绍下错位排列。

　　错位排列，顾名思义就是一个n元排列，每个元素不能排在自己的位置上的方案数，一般记作$D(n)$。

　　通项公式：

　　　　$$D_n=n!\times(1-\frac{1}{1!}+\frac{1}{2!}-\frac{1}{3!}…\frac{(-1)^n}{n!})$$

　　　证明：

　　　设$S$是由$1,2,…n$构成的所有全排列组成的集合，则$|S|=n!$。

　　　设$A_i$是在$1,2,…n$的所有排列种由第$i$个位置上的元素恰好是$i$的所有排列组成的集合，则有：$|A_i|=(n-1)!$。

　　　同理可得：$|A_i\cap A_j|=(n-2)!$

　　　……

　　　一般情况下有：$|A_{i1}\cap A_{i2}\cap …\cap A_{ik}|=(n-k)!$。

　　　因为$D_n$是$S$中不满足性质$P_1,P_2,…,P_n$的元素个数，所以由容斥原理的：

　　　$D_n=|\overline A_1\cap \overline A_2 …\cap \overline A_n|$

　　　　　$=n!-C(n,1)*(n-1)!+C(n,2)*(n-2)!-…(-1)^nC(n,n)*0!$

　　　　　$=n!\times(1-\frac{1}{1!}+\frac{1}{2!}-…\frac{(-1)^n}{n!})$

　　递推公式：

　　　　$$D_n=(n-1)\times(D_{n-1}+D_{n-2})$$

　　　证明：

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

　　　第二步，放编号为$k$的元素，这时有两种情况：(1)把它固定到位置$n$，由于第$n$个元素固定到了位置$k$，剩下$n-2$个元素就有$D_{n-2}$种方法；(2)第$k$个元素不能放到位置$n$，而第$n$个元素固定到了位置$k$，于是$n-1$个元素，有$D_{n-1}$种方法；

　　　综上得到$D_n = (n-1) \times(D_{n-2} + D_{n-1})$，特殊地，$D_1=0, D_2=1$。

　　当然更为常用的是后面的递推公式，比如本题。

　　不难发现本题答案为$C(n,m)\times D(n-m)$。

　　于是我们只要预处理出$10^6$内的阶乘取模、阶乘的逆元、错排的方案数就好了。

代码：

/*Code by 520 -- 9.14*/

#include<bits/stdc++.h>

#define il inline

#define ll long long

#define RE register

#define For(i,a,b) for(RE int (i)=(a);(i)<=(b);(i)++)

#define Bor(i,a,b) for(RE int (i)=(b);(i)>=(a);(i)--)

using namespace std;

const ll N=,mod=1e9+;

int n,m;

ll d[N],c[N],inv[N];

int gi(){

    int a=;char x=getchar();

    while(x<''||x>'')x=getchar();

    while(x>=''&&x<='')a=(a<<)+(a<<)+(x^),x=getchar();

    return a;

}

il void Pre(){

    d[]=,d[]=;

    For(i,,) d[i]=(i-)*(d[i-]+d[i-])%mod;

    c[]=,c[]=,inv[]=;

    For(i,,) c[i]=c[i-]*i%mod,inv[i]=(mod-mod/i*inv[mod%i]%mod)%mod;

    For(i,,) inv[i]=inv[i]*inv[i-]%mod;

}

int main(){

    Pre();

    int T=gi();

    while(T--) {

        n=gi(),m=gi();

        if(n-m==) printf("0\n");

        else if(n==m) printf("1\n");

        else if(!m) printf("%lld\n",d[n]);

        else printf("%lld\n",c[n]*inv[m]%mod*inv[n-m]%mod*d[n-m]%mod);

    }

    return ;

}