题目大意
a是关于n的排列。
问有多少符合条件的排列——恰有m个位置满足a[i]=i
错排
设dp[i]表示i个元素的错排方案数。
什么意思呢?就是关于i的排列其中不存在任意的a[j]=j。
答案显然就是Cmn∗dp[n−m]
组合数快速算可以预处理阶乘以及阶乘的逆元。
怎么预处理出dp呢?
我们可以使用容斥原理:
例如dp[n],其实就是要满足n个条件——第i个条件为a[i]!=i
那么合法方案数=至少不满足0个条件-至少不满足1个条件+至少不满足2个条件-至少不满足3个条件+……
也就是
dp[n]=∑ni=0(−1)i∗Cin∗(n−i)!
dp[n]=∑ni=0(−1)i∗n!i!∗(n−i)!∗(n−i)!
dp[n]=∑ni=0(−1)i∗n!i!
dp[n]=(−1)n+∑n−1i=0(−1)i∗n!i!
dp[n]=(−1)n+n∗∑n−1i=0(−1)i∗(n−1)!i!
dp[n]=(−1)n+n∗dp[n−1]
于是我们可以线性的预处理出dp数组。
然后这题就可以做了。
#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fd(i,a,b) for(i=a;i>=b;i--)
using namespace std;
typedef long long ll;
const int mo=1000000007,maxn=1000000+10;
int f[maxn],g[maxn],dp[maxn];
int i,j,k,l,t,n,m,ca,ans;
int qsm(int x,int y){
if (!y) return 1;
int t=qsm(x,y/2);
t=(ll)t*t%mo;
if (y%2) t=(ll)t*x%mo;
return t;
}
int main(){
f[0]=1;
fo(i,1,1000000) f[i]=(ll)f[i-1]*i%mo;
g[1000000]=qsm(f[1000000],mo-2);
fd(i,999999,0) g[i]=(ll)g[i+1]*(i+1)%mo;
dp[0]=1;dp[1]=0;
fo(i,2,1000000)
dp[i]=(((ll)dp[i-1]*i%mo+((i%2)?-1:1))%mo+mo)%mo;
scanf("%d",&ca);
while (ca--){
scanf("%d%d",&n,&m);
ans=(ll)f[n]*g[m]%mo*g[n-m]%mo*dp[n-m]%mo;
printf("%d\n",ans);
}
}