bzoj4517 [Sdoi2016]排列计数

Description

---

求有多少种长度为 n 的序列 A，满足以下条件：
1 ~ n 这 n 个数在序列中各出现了一次
若第 i 个数 A[i] 的值为 i，则称 i 是稳定的。序列恰好有 m 个数是稳定的
满足条件的序列可能很多，序列数对 10^9+7 取模。

T=500000
n≤1000000
m≤1000000

Solution

---

一开始想复杂了呀
等价于从n个数里挑m个稳定，剩下n-m个错排
组合数预处理一下阶乘和逆元，剩下的随便搞

Code

---

#include <stdio.h>
#include <string.h>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)

typedef long long LL;
const int MOD=1000000007;
const int N=2000005;

LL f[N],a[N];

LL ksm(LL x,LL dep) {
    if (dep==0) return 1;
    if (dep==1) return x;
    LL tmp=ksm(x,dep/2);
    if (dep%2) return tmp*tmp%MOD*x%MOD;
    return tmp*tmp%MOD;
}

int main(void) {
    f[0]=a[0]=1; f[1]=0; f[2]=1; a[1]=1;
    rep(i,3,N-1) f[i]=(f[i-1]+f[i-2])%MOD*(i-1)%MOD;
    rep(i,2,N-1) a[i]=a[i-1]*i%MOD;
    int T; scanf("%d",&T);
    while (T--) {
        int n,m; scanf("%d%d",&n,&m);
        LL ans=a[n]*ksm(a[m],MOD-2)%MOD*ksm(a[n-m],MOD-2)%MOD*f[n-m]%MOD;
        if (n==1&&m==1) ans=1;
        printf("%lld\n", ans);
    }
    return 0;
}