【uoj428】普通的計數題

Portal --> uoj428

Solution

　　不會胖子的一個log正解qwq只能慫慫滴寫分治了qwq

​　　首先就是一個我想不到的轉化qwq

​　　咱們將第\(i\)次操做加入的數當作一個編號爲\(i\)的節點的權值，而且把操做\(2\)中刪除的節點當作\(i\)號點的兒子，那麼整個操做序列對應一個樹結構，而且知足父親的編號大於兒子，特別的，原來的\(0\)對應了葉子節點，\(1\)對應了非葉子節點，那麼原題中的限制就變成了：當某個節點的兒子都是葉子的時候，它的兒子個數在\(B\)集合中，不然葉子兒子的個數在\(A\)集合中

​　　而後爲了方便一塊兒處理，咱們強行把\(0\)丟到\(B\)集合中，這樣葉子節點也知足條件了

​　　因而乎咱們就能夠dp了：記\(f[i]\)表示\(i\)個節點的知足條件的有根樹個數，\(g[i]\)表示\(i\)個節點的知足條件的森林個數，而且森林中每棵樹的節點個數很多於\(2\)，那麼dp的轉移方程爲：
\[ \begin{aligned} g[i]&=f[i]+\sum\limits_{j=2}^{i-1}g[i-j]\cdot f[j]\binom{i-1}{j-1}\\ f[i]&=[i-1\in B]+\sum\limits_{j=0}^{i-2}[j\in A]\binom{i-1}{j}g[i-1-j] \end{aligned} \]
​　　具體一點的話就是，\(g\)的轉移方程中\(j\)枚舉的是第\(i\)個點所在樹的大小，而後要從前面\(i-1\)個點裏面選\(j-1\)個出來和\(i\)號點組成一棵樹；\(f\)的轉移方程中前半部分是全部的兒子都是葉子的狀況，後半部分是枚舉葉子兒子的個數，而後\(g[i-1-j]\)表示的是剩下的非葉子兒子有多少種不一樣的方案

　　接下來正解對這個東西進行一些高級處理而後用牛頓迭代去搞了qwq

​　　然而我並不會因此就用分治

​　　注意到上面的式子其實已經能夠直接分治ntt了，把組合數拆一下而後按套路寫就行了

　　須要注意的事情是：由於這裏計算的\(g\)轉移跟本身有關，因此在分治完以後卷積算左邊對右邊貢獻的時候，可能會遇到須要用到的\(g\)尚未算出來（或者是須要用到的\(f\)）的狀況，卷積的時候就不會把貢獻算進去了，因此這個時候，若是說咱們當前的分治區間的左端點不是\(1\)，那麼就有可能包含了某個由於還沒被算出來而致使貢獻漏算的\(g\)或者\(f\)，因此咱們應該作兩次ntt，把漏算的貢獻加回去

​　　其實大概就是這樣的狀況：

​　　而後實現上就是。。當前分治區間的\(l\neq 1\)時，計算\(g\)的時候，枚舉\(g\)的下標在左半邊，作一次卷積；再枚舉\(f\)的下標在左半邊，作一次卷積，順序的話其實也會有一點影響，\(g\)下標在左半邊的那次卷積必需要保證用到的是尚未更新過右半邊的\(f\)，不然就會重複計算一些貢獻

​　　計算\(f\)的話由於並無與本身有關，因此直接算就行了不須要考慮那麼多

​　　一個小trick：注意到算\(g\)的時候咱們的\(j\)是從\(2\)開始枚舉的，爲了方便咱們能夠強行先令\(f[1]=0\)而後分治ntt，最後再把\(f[1]\)的值賦回去，這樣就中間寫的時候就能夠比較無腦了

​　　而後就十分愉快地作完了ovo（雖說是\(O(nlog^2n)\)作法qwq）

Code

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=114514+10,MOD=998244353;
int fac[N],invfac[N];
int inA[N],inB[N];
int f[N],g[N];
int n,m,lena,lenb;
int plu(int x,int y){return (1LL*x+y)-(1LL*x+y>=MOD?MOD:0);}
int mul(int x,int y){return 1LL*x*y%MOD;}
int ksm(int x,int y){
    int ret=1,base=x;
    for (;y;y>>=1,base=mul(base,base))
        if (y&1) ret=mul(ret,base);
    return ret;
}
namespace NTT{/*{{{*/
    const int N=(1<<18)+10,TOP=18,G=3;
    int A[N],B[N],W[N][2],rev[N];
    int len,invlen,invg;
    void prework(){
        invg=ksm(G,MOD-2);
        for (int i=1;i<=TOP;++i){
            W[1<<i][0]=ksm(G,(MOD-1)/(1<<i));
            W[1<<i][1]=ksm(invg,(MOD-1)/(1<<i));
        }
    }
    void get_len(int n){
        for (int i=0;i<len;++i) A[i]=B[i]=0;
        int bit=0;
        for (len=1;len<=n;len<<=1,++bit);
        rev[0]=0;
        for (int i=1;i<len;++i) rev[i]=(rev[i>>1]>>1)|((i&1)<<(bit-1));
        invlen=ksm(len,MOD-2);
    }
    void ntt(int *a,int op){
        int w,w_n,u,v;
        for (int i=0;i<len;++i) 
            if (rev[i]>i) swap(a[i],a[rev[i]]);
        for (int step=2;step<=len;step<<=1){
            w_n=W[step][op==-1];
            for (int st=0;st<len;st+=step){
                w=1;
                for (int i=0;i<(step>>1);++i){
                    v=mul(a[st+i+(step>>1)],w);
                    u=a[st+i];
                    a[st+i]=plu(u,v);
                    a[st+i+(step>>1)]=plu(u,MOD-v);
                    w=mul(w,w_n);
                }
            }
        }
        if (op==1) return;
        for (int i=0;i<len;++i) a[i]=mul(a[i],invlen);
    }
    void calc(){
        ntt(A,1);
        ntt(B,1);
        for (int i=0;i<len;++i) A[i]=mul(A[i],B[i]);
        ntt(A,-1);
    }
}/*}}}*/
void calc(int l,int r){
    int mid=l+r>>1,len=r-l+1,lenl=mid-l+1,lenr=r-mid;
    NTT::get_len(lenl+len);
    for (int i=l;i<=mid;++i) NTT::A[i-l]=mul(g[i],invfac[i]);
    for (int i=0;i<r-l;++i) NTT::B[i]=mul(inA[i],invfac[i]);
    NTT::calc();
    for (int i=mid+1;i<=r;++i) f[i]=plu(f[i],NTT::A[i-l-1]);//f has been updated

    NTT::get_len(lenl+len);
    for (int i=l;i<=mid;++i) NTT::A[i-l]=mul(f[i],invfac[i-1]);
    for (int i=1;i<=r-l;++i) NTT::B[i-1]=mul(g[i],invfac[i]);//should use the one which hasn't been updated
    NTT::calc();
    for (int i=mid+1;i<=r;++i) g[i]=plu(g[i],NTT::A[i-l-1]);

    if (l==1) return;
    NTT::get_len(lenl+len);
    for (int i=l;i<=mid;++i) NTT::A[i-l]=mul(g[i],invfac[i]);
    for (int i=1;i<=r-l;++i) NTT::B[i-1]=mul(f[i],invfac[i-1]);
    NTT::calc();
    for (int i=mid+1;i<=r;++i) g[i]=plu(g[i],NTT::A[i-l-1]);
}
void solve(int l,int r){
    if (l==r){
        if (l>1) //let f[1]=0 first: can't take f[1] into account during the dp
            f[l]=plu(inB[l-1],mul(f[l],fac[l-1]));
        g[l]=plu(f[l],mul(g[l],fac[l-1]));
        return;
    }
    int mid=l+r>>1;
    solve(l,mid);
    calc(l,r);
    solve(mid+1,r);
}
void prework(int n){
    fac[0]=1;
    for (int i=1;i<=n;++i) fac[i]=mul(fac[i-1],i);
    invfac[n]=ksm(fac[n],MOD-2);
    for (int i=n-1;i>=0;--i) invfac[i]=mul(invfac[i+1],i+1);
}

int main(){
#ifndef ONLINE_JUDGE
    freopen("a.in","r",stdin);
#endif
    int x;
    NTT::prework();
    scanf("%d%d%d",&n,&lena,&lenb);
    for (int i=1;i<=lena;++i) scanf("%d",&x),inA[x]=1;
    for (int i=1;i<=lenb;++i) scanf("%d",&x),inB[x]=1;
    inB[0]=1;
    prework(n);
    g[0]=1;
    solve(1,n);
    f[1]=1;
    printf("%d\n",f[n]);

    /*g[0]=1; f[1]=1;
    for (int i=2;i<=n;++i){
        g[i]=0; f[i]=0;
        for (int j=2;j<=i;++j)
            g[i]=plu(g[i],mul(f[j],mul(invfac[j-1],mul(g[i-j],invfac[i-j]))));
        g[i]=mul(g[i],fac[i-1]);
        for (int j=0;j<=i-2;++j)
            f[i]=plu(f[i],mul(inA[j],mul(invfac[j],mul(g[i-1-j],invfac[i-1-j]))));
        f[i]=plu(mul(f[i],fac[i-1]),inB[i-1]);
        g[i]=plu(g[i],f[i]);
    }
    printf("%d\n",f[n]);*/
}