洛谷P4071 排列计数

传送门：

P4071 [SDOI2016] 排列计数 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)https://www.luogu.com.cn/problem/P4071题干:

有多少个1到n的排列a，使得恰好有m个位置满足ai = i ， 回答T组询问，答案多 10^9+7取模

数据范围： n,m<=1e6, T<=1e5;

思路：

第一步：假设有m个位置满足ai=i，先从n个数字里面选出m个位置，然后把它们放到它们应该在的地方。

一共有C（n，m）种方案 。

第二步：求剩下的n-m个数字 错排的情况有多少种，

假设 d（x）表示 t个数字错排的情况，那么由公式可得：

d（x） = (x-1) ( d(x-1) + d(x-2) ) 

很明显满足递推关系。

所以答案就是： C（n，m）*  （n-m-1）* （ d(x-1)+d(x-2) )  % P

如何求组合数？ 我们都知道公式     

要求 整个式子模P的值怎么求呢？

第一步，预处理 所有的阶乘模上P的值，放进一个数组frc。

第二步，设置一个求逆元的函数inv（），由于模数是质数，所以我们考虑用快速幂。

又由于第一步已经求了所有的阶乘，所以第二步，我们用上面的结果就可，求出m！，（n-m）！

的逆元，然后相乘取模： inv（m！）*inv（（n-m）！） *  n！   （mod p）

第三步，预处理所有的错排情况，存入数组D。

(别忘了特判一下，n==m的情况）

最后美美滴写答案就可以了。

#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<cstdio>
#include<cmath>
#include<string>
#include<cstring>
#include<string>
#include<algorithm>
#include<vector>
#include<cctype>
#include<map>
#include<set>
#include<queue>
#include<numeric>
#include<iomanip>
using namespace std;


typedef long long ll;
const ll N = 1e6+7;
const ll M = 1e6+7;
const ll P = 1e9 + 7;
ll get_inv(ll a, ll b) {
	ll s = 1;
	while (b) {
		if (b & 1)s = s * a % P;
		a = a * a % P;
		b >>= 1;
	}
	return s;
}
ll frc[N];
void get_frc(void) {
	
	for (ll i = 1; i <= N; i++) {
		frc[i] = frc[i - 1] * i % P;
	}
}
ll D[N];
void get_d() {
	for (ll i = 3; i <= N; i++) {
		D[i] = ((i - 1)) * (D[i - 1] + D[i - 2]) % P;
	}
}

int main() {
	frc[0] = 1;
	D[1] = 0;
	D[2] = 1;
	get_d();
	get_frc();
	ll t;
	cin >> t;
	while (t--) {
		ll n, m;
		cin >> n >> m;
		 if (n == m)cout << 1 << endl;
		else {
			ll sum = get_inv(frc[m], P - 2)%P  * get_inv(frc[n - m], P - 2)%P  * frc[n]%P  * D[n - m] % P;
			cout << sum<<endl;
		}
	}
}