[bzoj5287] [HNOI2018]毒瘤

题目描述

从前有一名毒瘤。

毒瘤最近发现了量产毒瘤题的奥秘。考虑如下类型的数据结构题:给出一个数组,要求支持若干种奇奇怪怪的修改操作(比如区间加一个数,或者区间开平方),并支持询问区间和。毒瘤考虑了n个这样的修改操作,并编号为\(1\sim n\)。当毒瘤要出数据结构题的时候,他就将这些修改操作中选若干个出来,然后出成一道题。

当然了,这样出的题有可能不可做。通过精妙的数学推理,毒瘤揭露了这些修改操作的关系:有m对“互相排斥”的修改操作,第i对是第ui个操作和第vi个操作。当一道题同时含有ui和vi这两个操作时,这道题就会变得不可做。另一方面,一道题中不包含任何“互相排斥”的修改操作时,这个题就是可做的。此外,毒瘤还发现了一个规律:m-n是一个很小的数字,且任意两个修改操作都是连通的。两个修改操作a,b是连通的,当且仅当存在若干操作\(t_0,t_1,...,t_l\),使得\(t_0=a,t_l=b\),且对1≤i≤l,\(t_{i-1}\)\(t_i\)都是“互相排斥”的修改操作。

一堆“互相排斥”的修改操作称为互斥对。现在毒瘤想知道,给定值n和m个互斥对,他共能出出多少道可做的不同的数据结构题。两道数据结构题是不同的,当且仅当有一个修改操作在其中一道题中存在,而在另一道题中不存在。

输入输出格式

输入格式:

第一行为正整数n,m。

接下来m行,每行两个正整数u,v,代表一对“互相排斥”的修改操作。

输出格式:

输出一行一个整数,代表毒瘤可以出的可做的不同的“互相排斥”的修改操作的个数。这个数可能很大,所以只输出模998244353后的值。

Solution

虚树。

先处理出一颗生成树,考虑到非树边很少,考虑暴力枚举非树边两端的状态,复杂度\(O(4^{n-m+1}n)\)

然后优化一下,对于一条非树边,两端的状态只需要枚举\((1,0)\)\((0,1)\)就好了,\((1,1)\)显然不合法,\((0,0)\)可以在\(dp\)的时候得到。复杂度\(O(2^{n-m+1}n)\)

考虑到上面的过程枚举时,树边是不变的,也就是说,可以考虑把非树边所在的点建出一颗虚树,然后对于虚树上的点,\(dp\)方程一定可以表示成:
\[ f_{u,0/1}=\prod_{v\in son_u}k_{0/1,0}f_{v,0}+k_{0/1,1}f_{v,1} \]
其中\(k\)为固定的系数,对于虚树上每条边,这个转移都是固定的。

所以可以\(O(n)\)先在原树\(dp\)出系数,然后暴力枚举关键点状态暴力虚树上\(dp\)就好了。

复杂度\(O(n+2^{2(m-n+1)}(n-m+1))\),足以通过此题。

#include<bits/stdc++.h>
using namespace std;

#define int long long 

void read(int &x) {
    x=0;int f=1;char ch=getchar();
    for(;!isdigit(ch);ch=getchar()) if(ch=='-') f=-f;
    for(;isdigit(ch);ch=getchar()) x=x*10+ch-'0';x*=f;
}
 
void print(int x) {
    if(x<0) putchar('-'),x=-x;
    if(!x) return ;print(x/10),putchar(x%10+48);
}
void write(int x) {if(!x) putchar('0');else print(x);putchar('\n');}

const int mod = 998244353;
const int maxn = 3e5+10;

int n,m,s[100],tt[100],cnt,sz[maxn],dfn[maxn],dep[maxn],use[maxn],val[maxn],vis[maxn],g[maxn][2],pr[maxn];

struct data {int k[2][2];}epsilon;

struct Dsu {
    int fa[maxn];
    void init() {for(int i=1;i<=n;i++) fa[i]=i;}
    int find(int x) {return fa[x]==x?x:fa[x]=find(fa[x]);}
}dsu;

struct Input_Tree {
    int head[maxn],tot,f[maxn][20],dfn_cnt;
    struct edge{int to,nxt;}e[maxn<<1];

    void add(int u,int v) {e[++tot]=(edge){v,head[u]},head[u]=tot;}
    void ins(int u,int v) {add(u,v),add(v,u);}

    void dfs(int x,int fa) {
        sz[x]=1,dfn[x]=++dfn_cnt,dep[x]=dep[fa]+1,f[x][0]=fa;
        for(int i=1;i<=19;i++) f[x][i]=f[f[x][i-1]][i-1];
        for(int i=head[x];i;i=e[i].nxt)
            if(e[i].to!=fa) dfs(e[i].to,x),sz[x]+=sz[e[i].to];
    }

    int lca(int x,int y) {
        if(dep[x]<dep[y]) swap(x,y);
        for(int i=19;~i;i--) if(dep[f[x][i]]>=dep[y]) x=f[x][i];
        if(x==y) return x;
        for(int i=19;~i;i--) if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
        return f[x][0];
    }

    void dp(int x,int fa) {
        int bo=1;g[x][0]=g[x][1]=1;
        for(int i=head[x];i;i=e[i].nxt)
            if(e[i].to!=fa) {
                bo=0,dp(e[i].to,x);int v=e[i].to;
                g[x][0]=1ll*g[x][0]*(g[v][0]+g[v][1])%mod;
                g[x][1]=1ll*g[x][1]*g[v][0]%mod;
            }
        if(bo) g[x][0]=g[x][1]=1;
    }

    int ns[maxn][2],nt[maxn][2];
    
    data get(int x,int fa) {
        ns[x][0]=1,ns[x][1]=0;
        nt[x][0]=0,nt[x][1]=1;
        while(x!=fa) {
            pr[x]=1;
            int pre=x;x=f[x][0];ns[x][0]=ns[x][1]=nt[x][0]=nt[x][1]=1;
            ns[x][0]=1ll*ns[x][0]*(ns[pre][0]+ns[pre][1])%mod;
            ns[x][1]=1ll*ns[x][1]*ns[pre][0]%mod;
            nt[x][0]=1ll*nt[x][0]*(nt[pre][0]+nt[pre][1])%mod;
            nt[x][1]=1ll*nt[x][1]*nt[pre][0]%mod;
            if(x==fa) break;
            for(int i=head[x];i;i=e[i].nxt)
                if(e[i].to!=f[x][0]&&e[i].to!=pre) {
                    int v=e[i].to;
                    ns[x][0]=1ll*ns[x][0]*(g[v][0]+g[v][1])%mod;
                    ns[x][1]=1ll*ns[x][1]*g[v][0]%mod;
                    nt[x][0]=1ll*nt[x][0]*(g[v][0]+g[v][1])%mod;
                    nt[x][1]=1ll*nt[x][1]*g[v][0]%mod;
                }
        }
        data ans;pr[fa]=1;
        ans.k[0][0]=ns[x][0],ans.k[1][0]=ns[x][1];
        ans.k[0][1]=nt[x][0],ans.k[1][1]=nt[x][1];
        return ans;
    }
}T;

int cmp(int x,int y) {return dfn[x]<dfn[y];}

struct Virtual_Tree {
    int head[maxn],tot;
    struct edge{int to,nxt;data k;}e[maxn<<1];

    void add(int u,int v) {e[++tot]=(edge){v,head[u],epsilon},head[u]=tot;}
    void ins(int u,int v) {add(u,v),add(v,u);}
    
    int in[maxn],k,use[maxn],used,sta[maxn],top,ban[maxn],f[maxn][2],val[maxn][2];

    void build() {
        sort(in+1,in+k+1,cmp);k=unique(in+1,in+k+1)-in-1;
        sta[++top]=1;
        for(int i=1;i<=k;i++) {
            if(in[i]==1) continue;
            int t=T.lca(in[i],sta[top]),pre=-1;
            while(dfn[sta[top]]>dfn[t]&&dfn[sta[top]]<dfn[t]+sz[t]) {
                if(pre!=-1) ins(sta[top],pre);
                pre=sta[top],top--;
            }
            if(pre!=-1) ins(t,pre);
            if(sta[top]!=t) sta[++top]=t;
            sta[++top]=in[i];
        }
        int pre=-1;
        while(top) {
            if(pre!=-1) ins(sta[top],pre);
            pre=sta[top],top--;
        }
    }

    void dfs(int x,int fa) {
        use[x]=1;
        for(int i=head[x];i;i=e[i].nxt)
            if(e[i].to!=fa) dfs(e[i].to,x);
    }

    void make(int x,int fa) {
        for(int i=head[x];i;i=e[i].nxt)
            if(e[i].to!=fa) {
                e[i].k=T.get(e[i].to,x);
                make(e[i].to,x);
            }
    }

    void dp(int x,int fa) {
        f[x][0]=val[x][0],f[x][1]=val[x][1];
        for(int i=head[x];i;i=e[i].nxt)
            if(e[i].to!=fa) {
                dp(e[i].to,x);data u=e[i].k;int v=e[i].to;
                f[x][0]=1ll*f[x][0]*(f[v][0]*u.k[0][0]%mod+f[v][1]*u.k[0][1]%mod)%mod;
                f[x][1]=1ll*f[x][1]*(f[v][0]*u.k[1][0]%mod+f[v][1]*u.k[1][1]%mod)%mod;
            }
        if(ban[x]==0) f[x][1]=0;
        else if(ban[x]==1) f[x][0]=0;
    }
    
    void get_val(int x,int fa) {
        val[x][0]=val[x][1]=1;
        for(int i=T.head[x],v=T.e[i].to;i;i=T.e[i].nxt,v=T.e[i].to)
            if(!pr[v]) {
                val[x][0]=1ll*val[x][0]*(g[v][0]+g[v][1])%mod;
                val[x][1]=1ll*val[x][1]*g[v][0]%mod;
            }
        for(int i=head[x];i;i=e[i].nxt)
            if(e[i].to!=fa) get_val(e[i].to,x);
    }
    
    void solve() 
        for(int i=1;i<=cnt;i++)
            in[++k]=s[i],in[++k]=tt[i],vis[s[i]]=1,vis[tt[i]]=1;
        build();make(1,0);int ans=0;
        memset(ban,-1,sizeof ban);

        get_val(1,0);

        for(int st=0;st<(1<<k);st++) {
            for(int i=1;i<=k;i++)
                if(st&(1<<(i-1))) ban[in[i]]=1;
                else ban[in[i]]=0;
            for(int i=1;i<=cnt;i++)
                if(ban[s[i]]==1&&ban[tt[i]]==1) goto loop;
            dp(1,0);
            ans=(0ll+ans+f[1][0]+f[1][1])%mod;
        loop:;
        }
        write(ans);
    }
}VT;

signed main() {
    read(n),read(m);dsu.init();
    for(int i=1,x,y;i<=m;i++) {
        read(x),read(y);int u=dsu.find(x),v=dsu.find(y);
        if(u==v) s[++cnt]=x,tt[cnt]=y;
        else dsu.fa[u]=v,T.ins(x,y);
    }
    T.dfs(1,0),T.dp(1,0);VT.solve();
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/hbyer/p/10243535.html