BZOJ 4517: [Sdoi2016]排列计数 [容斥原理]

4517: [Sdoi2016]排列计数

题意：多组询问，n的全排列中恰好m个不是错排的有多少个

---

容斥原理强行推♂倒她
$恰好m个不是错排 $
\[ =\ \ge m个不是错排 - \ge m+1个不是错排\binom{m+1}{m} - \ge m+2个不是错排\binom{m+2}{m}... \\ = \sum_{i=m}^n \binom{n}{i} (n-i)!\binom{i}{m} \\ = \frac{n!}{m!} \sum_{i=m}^n (-1)^{i-m} \frac{1}{(i-m)!} \]
预处理阶乘逆元前缀和就可以\(O(1)\)回答了
其实错排公式也是这么推倒来的

PS:发现题解全都是用的错排公式，等出一道你们不知道公式的题你们再用啊

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
using namespace std;
typedef long long ll;
#define fir first
#define sec second
const int N=1e6+5, P=1e9+7;
inline int read() {
    char c=getchar(); int x=0, f=1;
    while(c<'0' || c>'9') {if(c=='-')f=-1; c=getchar();}
    while(c>='0' && c<='9') {x=x*10+c-'0'; c=getchar();}
    return x*f;
}

int n, m;
ll inv[N], fac[N], facInv[N], s[N];
int main() {
    freopen("permutation.in","r",stdin);
    freopen("permutation.out","w",stdout);
    inv[1]=1; fac[0]=facInv[0]=1;
    s[0]=1;
    for(int i=1; i<N; i++) {
        if(i!=1) inv[i] = (P-P/i)*inv[P%i]%P;
        fac[i] = fac[i-1]*i%P;
        facInv[i] = facInv[i-1]*inv[i]%P;
        s[i] = (s[i-1] + ((i&1) ? -facInv[i] : facInv[i]))%P;
    }
    int T=read();
    while(T--) {
        n=read(); m=read();  
        ll ans = fac[n]*facInv[m]%P * s[n-m]%P; 
        if(ans<0) ans+=P;
        printf("%lld\n", ans);
    }
}