CF715E Complete the Permutations(第一类斯特林数)

题目

CF715E Complete the Permutations

作法

先考虑无\(0\)排列的最小花费，其实就是沿着置换交换，花费：\(n-\)环个数，因此咱们主要是要求出规定环的个数

考虑连边\(a_i\rightarrow b_i\)(仅非零数有出边)，自己造成环的无论(也没办法管)，考虑一条除首尾外都不为\(0\)，首尾至少有一个\(0\)的链(链)，那么还有三类：

\(0\rightarrow0(0(a)\)对应的\(b\)是\(0\)或\(b\)的最终点是\(0)\)

\(0\rightarrow x(0(a)\)对应的\(b\)的终点\(x\)无出边\()\)

\(x\rightarrow 0(x(a)\)对应的\(b\)为\(0\)或\(b\)的最终点是\(0)\)

记这三类为个数\(a\)，\(b\)，\(a\)；且\(b\)和\(c\)所构成的环得经过\(a\)间接接通

\(f_i\)为第二类本身和本身匹配所造成的环个数为\(i\)个的方案数：
\[f_i=\sum\limits_{j=i}^b\begin{bmatrix}j\\i\end{bmatrix}C_b^j(a+b-j)^{\underline {b-i}}\]

理解：从\(b\)条二类链选出\(j\)条，组成\(i\)个环，剩余的二类链随机匹配，\(x\rightarrow 0\)本身与本身连也是\(x\rightarrow 0\)，与一类就造成\(0\rightarrow0\)，其实最后剩余的二类链就会造成所有变成一类链，而一类链的数量并不会增多

\(g_i\)为三类的方案数，构造方法更上方相同，理解也同样

最后咱们仅剩三类了，此时数量仍是\(c\)，而其中一些三类得贡献出去给一二类成环，而消耗完了的这些就本身给本身成环就行，一类边最后还要排列一下：
\[h_i=\begin{bmatrix}a\\i\end{bmatrix}a!\]

咱们获得的就是一个三式卷积形式，暴力作

Code

#include<bits/stdc++.h>
typedef int LL;
const LL maxn=309,mod=998244353;
inline LL Read(){
    LL x(0),f(1); char c=getchar();
    while(c<'0' || c>'9'){
        if(c=='-') f=-1; c=getchar();
    }
    while(c>='0' && c<='9'){
        x=(x<<3)+(x<<1)+c-'0'; c=getchar();
    }
    return x*f;
}
LL n,a,b,c,d;
LL visit1[maxn],visit[maxn],nxt[maxn],a1[maxn],b1[maxn],ans[maxn],ru[maxn],fac[maxn];
LL f[maxn],g[maxn],h[maxn];
LL strl1[maxn][maxn],C[maxn][maxn],D[maxn][maxn];
void Dfs(LL x){
    if(!x) return;
    visit1[x]=true;
    visit[x]=true;
    if(nxt[x]!=x){
        if(visit1[nxt[x]]) ++d;
        else{
            Dfs(nxt[x]); nxt[x]=nxt[nxt[x]];
        }
    }
    visit1[x]=0;
}
int main(){
    n=Read();
    for(LL i=1;i<=n;++i) a1[i]=Read();
    for(LL i=1;i<=n;++i) b1[i]=Read();
    for(LL i=1;i<=n;++i) nxt[i]=i;
    for(LL i=1;i<=n;++i){
        if(a1[i]){
            if(a1[i]==b1[i]){
                ++d; visit[a1[i]]=true;
            }
            else{
                nxt[a1[i]]=b1[i];
            }
        }
        if(b1[i]) ++ru[b1[i]];
    }
    for(LL i=1;i<=n;++i) if(!visit[i]) Dfs(i);
    for(LL i=1;i<=n;++i)
        if(!a1[i]){
            if(nxt[b1[i]]) ++b;
            else ++a;
        }else
            if(!ru[a1[i]] && !nxt[b1[i]]) ++c;
    
    strl1[0][0]=strl1[1][1]=1;
    for(LL i=2;i<=n;++i)
        for(LL j=1;j<=i;++j)
            strl1[i][j]=(strl1[i-1][j-1]+1ll*(i-1)*strl1[i-1][j])%mod;
    C[0][0]=1;
    for(LL i=1;i<=n;++i){
        for(LL j=1;j<=i;++j)
            C[i][j]=1ll*(C[i-1][j-1]+C[i-1][j])%mod;
        C[i][0]=1;
    }
    D[0][0]=1;
    for(LL i=1;i<=n;++i){
        D[i][0]=1;
        for(LL j=1;j<=i;++j)
            D[i][j]=1ll*D[i][j-1]*(i-j+1)%mod;
    }
    fac[0]=fac[1]=1;
    for(LL i=2;i<=n;++i) fac[i]=1ll*fac[i-1]*i%mod;
    for(LL i=0;i<=b;++i)
        for(LL j=i;j<=b;++j)
            f[i]=(f[i]+1ll*strl1[j][i]*C[b][j]%mod*(a+b-j-1>=0?D[a+b-j-1][b-j]:1)%mod)%mod;
    for(LL i=0;i<=c;++i){
        for(LL j=i;j<=c;++j)
            g[i]=(g[i]+1ll*strl1[j][i]*C[c][j]%mod*(a+c-j-1>=0?D[a+c-j-1][c-j]:1)%mod)%mod;
    }
    for(LL i=0;i<=a;++i)
        h[i]=1ll*strl1[a][i]*fac[a]%mod;
    for(LL i=0;i<=n;++i)
        for(LL j=0;j<=i;++j)
            for(LL k=0;k<=i-j;++k)
                ans[i]=(ans[i]+1ll*h[j]*f[k]%mod*g[i-j-k]%mod)%mod;
    for(LL i=0;i<n;++i){
        if(n-i-d<0) printf("0 ");
        else printf("%d ",ans[n-i-d]);
    }
    return 0;
}