BZOJ4517: [Sdoi2016]排列计数(组合数+错位排列)

Time Limit: 60 Sec Memory Limit: 128 MB
Submit: 1626 Solved: 994
[Submit][Status][Discuss]

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上传

啊啊为什么我这么菜QWQ。。

这题一个公式就过去了，

考虑一个数$i$，只有当$i$在第$i$个位置时才能产生贡献，

那么需要产生$m$个数的方案就是$C_n^m$

然后让剩下的数错排，设错排的方案数为$D(i)$

递推公式$D[i]=(i-1)*D(i-1)*D(i-2)$

证明：

#include<cstdio>

#define int long long

using namespace std;

const int mod=1e9+;

const int MAXN=1e6+;

inline int read()

{

    char c=getchar();int x=,f=;

    while(c<''||c>''){if(c=='-')f=-;c=getchar();}

    while(c>=''&&c<=''){x=x*+c-'';c=getchar();}

    return x*f;

}

int fac[MAXN],D[MAXN],inv[MAXN];

void Pre()

{

    fac[]=fac[]=inv[]=inv[]=D[]=D[]=;

    for(int i=;i<=;i++) fac[i]=(i*fac[i-])%mod;

    for(int i=;i<=;i++) inv[i]=(mod-mod/i)*inv[mod%i]%mod;

    for(int i=;i<=;i++) inv[i]=(inv[i]*inv[i-])%mod;

    for(int i=;i<=;i++) D[i]=((i-)*(D[i-]+D[i-]))%mod;

}

int Query(int N,int M)

{

    return fac[N] %mod * inv[ N-M  ] %mod * inv[ M ] %mod * D[N-M] %mod;

}

main()

{

    Pre();

    int T=read();

    while(T--)

    {

        int N=read(),M=read();

        printf("%lld\n",Query(N,M)%mod);

    }

    return ;

}

巴特西