SDOI 2016 排列计数

SDOI 2016 排列计数

题目链接

设满足题目条件且长度为n恰好有m个数稳定的序列数为$ans(n,m)$
那么易得题目答案ANS可表示为：
$$ANS=ans(n,m)=C(n,m)\ast ans(n-m,0)$$
因为题目规定$n,m\le 1e6$，所以C(n,m)可以通过预处理阶乘+乘法逆元快速求得。
至于ans(n-m,0)，又被称作为全错位排位数。
即n个相异的元素排成一排且$a_i\ne i$的排列个数。
设长度为n的全错位排列数为F_n,有如下递推公式：
$$F_n=(n-1)(F_{n-1}+F_{n-2}),n\ge 3$$，其中$F_1=0,F_2=1$。另外对于长度为0的序列，规定其全错位排列数为1。

证明：依据递推的思想，显然$F_1=0,F_2=1$,对于$n\ge 3$，不妨设第n个数排在了第$k(k\ne n)$位。
1、若第k个数排在了第n位，那么对于n和k两个位置满足$a_n\ne n,a_k=n\ne k$，那么只剩下了n-2个数，显然剩下n-2个数可构成的全错位排列数为$F_{n-2}$
2、若第k个数没有排在第n位，则规定了$a_n\ne k$，那么可以将第n位假想为第k位，那么问题就转换成了n-1个数的全错位排列问题，解为$F_{n-1}$。
由于$1\le k<n$，所以$F_n=(n-1)(F_{n-1}+F_{n-2})$

接下来利用容斥原理来推出$F_n$的通项公式：

设n个数组成的全排列集合为I，则$|I|=n!$。设满足$a_i=i$的排列的集合为$A_i(1\le i\le n)$。有$|A_i|=(n-1)!$，$|A_i\cap A_j|=(n-2)!(1\le i<j\le n)$，$|A_i\cap A_{i+1}\cap ..\cap A_{i+k}|=(n-k-1)!$
有
$\begin{array}{rl}{F}_n& =|I|-|A_1\cup A_2\cup ...\cup A_n|\\ & =|I|-(C_n^1|A_1|-C_n^2|A_1\cap A_2|+C_n^3|A_1\cap A_2\cap A_3|+..+(-1{)}^{n-1}|A_1\cap A_2\cap ..\cap A_n|)\\ & =n!-C_n^1(n-1)!+C_n^2(n-2)!+...+(-1{)}^i{C}_n^i(n-i)!+..+(-1{)}^n{C}_n^{n}0!\\ & =n!\sum _{i=0}^n(-1{)}^i\frac{1}{i!}\end{array}$

从而也可以验证递推公式：
$$\begin{array}{rl}{F}_{n+1}& =n\cdot (F_n+F_{n-1})\\ & =n\cdot (n!\sum _{i=0}^n(-1{)}^i\frac{1}{i!}+(n-1)!\sum _{i=0}^{n-1}(-1{)}^i\frac{1}{i!})\\ & =n\cdot (n!\sum _{i=0}^{n-1}(-1{)}^i\frac{1}{i!}+(-1{)}^n+(n-1)!\sum _{i=0}^{n-1}(-1{)}^i\frac{1}{i!}\\ & =(n+1)!\sum _{i=0}^{n-1}(-1{)}^i\frac{1}{i!}+(-1{)}^n\cdot n\\ & =(n+1)!\cdot (\sum _{i=0}^{n-1}(-1{)}^i\frac{1}{i!}+(-1{)}^n\cdot \frac{n}{(n+1)!})\\ & =(n+1)!\cdot (\sum _{i=0}^{n-1}(-1{)}^i\frac{1}{i!}+(-1{)}^n\cdot \frac{n+1-1}{(n+1)!})\\ & =(n+1)!\cdot (\sum _{i=0}^{n-1}(-1{)}^i\frac{1}{i!}+(-1{)}^n\cdot \frac{1}{n!}+(-1{)}^{n+1}\frac{1}{(n+1)!})\\ & =(n+1)!\cdot \sum _{i=0}^{n+1}(-1{)}^i\frac{1}{i!}\\ & =F_{n+1}\end{array}$$

代码如下：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int max_n=1e6+5;
const int mod=1e9+7;
ll factorial[max_n];
ll f[max_n];
ll rev[max_n];
void init(void)
{
	memset(rev,-1,sizeof(rev));
	factorial[0]=1;
	f[0]=1;
	f[1]=0;f[2]=1;
	for(ll i=1;i<max_n;i++)
	factorial[i]=factorial[i-1]*i%mod;
	for(ll i=3;i<max_n;i++)
	f[i]=((i-1)*(f[i-1]+f[i-2]))%mod;
}
ll qpow(ll index,ll n,ll p)
{
	if(rev[index]!=-1)return rev[index];
	ll a=factorial[index];
	ll ans=1;
	while(n)
	{
		if(n&1)ans=ans*a%p;
		a=a*a%p;
		n>>=1;
	}
	return rev[index]=ans;
}
ll C(ll n,ll m)
{
	if(m>n)return 0;
	return factorial[n]*qpow(m,mod-2,mod)%mod*qpow(n-m,mod-2,mod)%mod;
}
int main(void)
{
	int t;
	init();
	scanf("%d",&t);
	ll n,m;
	while(t--)
	{
		scanf("%lld%lld",&n,&m);
		printf("%lld\n",C(n,m)*f[n-m]%mod);
	}
}
//5
//1 0
//1 1
//5 2
//100 50
//10000 5000