传送门
Description
给出一张 \(n\)个点 \(m\)条边的无向连通图,其中每条边至多属于一个简单环,保证没有自环,可能有重边。你需要为其中每条边定向,其中第 \(i\)个点的出度不能超过\(a_i\) ,求方案数(对\(998244353\)取模)。
Solution
建出原图的圆方树,其中两个点和一条边组成的不算做一个双联通分量,圆方树的的根是\(1\)
定义\(f[x][i]\)表示,第\(x\)个点,它上面的点给它的入度为\(i\),以\(x\)为根的子树内给边定向的方案数。
更清楚一点:
如果\(x\)是一个圆点,且它的父亲也是圆点
\(f[x][0/1]\)即表示它的父亲节点与它之间的边是否是指向\(x\)的
如果\(x\)是一个方点,且它的父亲是圆点
\(f[x][0/1/2]\)即表示这个环的最上面的两条边(就是于\(fa\)直接相连的那两条),不指向\(fa\)的数量(\(0/1/2\))
如果\(x\)是一个圆点,且它的父亲是方点
\(f[x][0/1/2]\)表示的就是它所在的环上与它直接相邻的两条边,指向它自己的边的数量\((0/1/2)\)
\(dfs\)时考虑分情况转移:
\(x\)是圆点,显然有:
\[ f[x][0]=\sum_{k_1+K_2+...+k_p\leq a[x]}\prod_{v_i} f[v_i][Sum_{v_i}-k_i] \]
其中,如果\(v_i\)是方点,\(Sum_{v_i}=2\),如果\(v_i\)是圆点,\(Sum_{v_i}=1\)因为是一个背包的形式,可以直接利用\(NTT\)加速
\(x\)是方点,设\(x\)的父亲节点是\(y\)
枚举环上与\(y\)相连的一条边的方向,然后顺着环递推即可
最后,在进行分治\(NTT\)时,每次都把两个数组全部清空复杂度是不对的
每次做完后,把当前的数组有修改的地方清为\(0\)就行
代码写得真的很清楚啦
Code
#include<bits/stdc++.h>
#define min(a,b) ((a)<(b)?(a):(b))
namespace IO
{
const int lim=(1<<20)+5;
char buf[lim+5],*S,*T;
inline char gc(){if(S==T){T=(S=buf)+fread(buf,1,lim,stdin);if(S==T)return EOF;}return *S++;}
inline int read()
{
int x;char ch;bool f;
for(f=0;(ch=gc())<'0'||ch>'9';f=ch=='-');
for(x=ch^'0';(ch=gc())>='0'&&ch<='9';x=(x<<1)+(x<<3)+(ch^'0'));
return f?-x:x;
}
}
#define reg register
#define son e[i].to
const int MN=1e5+5,P=998244353,g=3,invg=332748118,MX=524288;
int u[MX],v[MX],N;
void Add(int &x,int y){x+=y;if(x>=P)x-=P;}
class MyNTT
{
int pos[MX],invN,di;
int fpow(int x,int m){int r=1;for(;m;m>>=1,x=1ll*x*x%P)if(m&1)r=1ll*r*x%P;return r;}
void NTT(int *a,bool type)
{
reg int w,wn,i,j,p,k,X,Y;
for(i=0;i<N;++i) if(i<pos[i]) std::swap(a[pos[i]],a[i]);
for(i=1;i<N;i<<=1)
{
wn=fpow(type?g:invg,(P-1)/(i<<1));
for(p=i<<1,j=0;j<N;j+=p)
for(w=1,k=0;k<i;++k,w=1ll*w*wn%P)
{
X=a[j+k];Y=1ll*a[i+j+k]*w%P;
a[j+k]=(X+Y)%P;a[i+j+k]=(X-Y+P)%P;
}
}
if(!type)for(i=0;i<N;++i)a[i]=1ll*invN*a[i]%P;
}
public:
void Pro(int *a,int *b,int n,int m)
{
for(di=0,N=1;N<n+m;N<<=1,++di);invN=fpow(N,P-2);
reg int i;
for(i=0;i<N;++i)pos[i]=(pos[i>>1]>>1)|((i&1)<<(di-1));
NTT(a,1);NTT(b,1);
for(i=0;i<N;++i)a[i]=1ll*a[i]*b[i]%P;
NTT(a,0);
}
}ntt;
int pin,t;
struct Node{int x,nex;}O[MN*40];
class MyCDQ
{
int res[MN];
public:
struct poly
{
int hr,n;
poly(int _=0,int __=-1):hr(_),n(__){}
void push_back(int x){++n;O[++pin]=(Node){x,hr};hr=pin;}
void To_array(int *a){for(int i=hr,j=n;i;i=O[i].nex)a[j--]=O[i].x;}
}p[MN];
poly Cdq(int l,int r)
{
if(l==r) return p[l];
reg int mid=(l+r)>>1;
poly L=Cdq(l,mid),R=Cdq(mid+1,r);
L.To_array(u);R.To_array(v);
ntt.Pro(u,v,L.n+1,R.n+1);
poly ans;for(reg int i=0;i<L.n+R.n+1;++i)ans.push_back(u[i]);
for(reg int i=0;i<N;++i) u[i]=0;
for(reg int i=0;i<N;++i) v[i]=0;
return ans;
}
void clear(){t=0;}
void add(){++t;p[t].hr=0;p[t].n=-1;}
void pb(int x){p[t].push_back(x);}
void get_ans(int &_0,int &_1,int &_2,int limi)
{
poly c=Cdq(1,t);
c.To_array(res);
for(reg int i=c.n+1;i<=limi;++i) res[i]=0;
for(reg int i=1;i<=limi;++i) Add(res[i],res[i-1]);
_0=res[limi];if(limi)_1=res[limi-1];if(limi>1)_2=res[limi-2];
}
}cdq;
int n,m,du[MN],hr[MN],Hr[MN<<1],en=1,A[MN],f[MN<<1][3];
struct edge{int to,nex;}e[MN<<3];
inline void ins(int x,int y,int *h){e[++en]=(edge){y,h[x]};h[x]=en;}
int dfn[MN],low[MN],st[MN],tp,ind,num;
void tj(int x,int F=0)
{
dfn[x]=low[x]=++ind;st[tp++]=x;
reg int i;
for(i=hr[x];i;i=e[i].nex)if(i^F^1)
{
if(!dfn[son])
{
tj(son,i);
low[x]=min(low[x],low[son]);
if(dfn[x]==low[son])
{
++num;ins(x,num,Hr);
for(;st[tp]!=son;--tp) ins(num,st[tp-1],Hr);
}
else if(low[son]>dfn[x]) --tp,ins(x,son,Hr);
}
else low[x]=min(low[x],dfn[son]);
}
}
void dfs(int x)
{
reg int i,j,_;
for(i=Hr[x];i;i=e[i].nex) dfs(son);
if(x<=n)
{
if(!Hr[x]) f[x][0]=1,f[x][1]=A[x]>0,f[x][2]=A[x]>1;
else
{
cdq.clear();
for(i=Hr[x];i;i=e[i].nex)
{
cdq.add();
if(son>n) cdq.pb(f[son][2]);
cdq.pb(f[son][1]);cdq.pb(f[son][0]);
}
cdq.get_ans(f[x][0],f[x][1],f[x][2],A[x]);
}
}
else
{
for(_=0;_<2;++_)
{
int S0[2]={0,0},S1[2];S0[_]=1;
for(i=Hr[x];i;i=e[i].nex)
{
S1[0]=(1ll*S0[0]*f[son][1]+1ll*S0[1]*f[son][2])%P;
S1[1]=(1ll*S0[0]*f[son][0]+1ll*S0[1]*f[son][1])%P;
S0[0]=S1[0];S0[1]=S1[1];
}
Add(f[x][_+1],S0[0]);
Add(f[x][_],S0[1]);
}
}
}
int main()
{
num=n=IO::read();m=IO::read();
reg int i,x,y;
for(i=1;i<=m;++i) x=IO::read(),y=IO::read(),ins(x,y,hr),ins(y,x,hr),++du[x],++du[y];
for(i=1;i<=n;++i) A[i]=IO::read(),A[i]=min(A[i],du[i]);
for(i=1;i<=n;++i) if(!dfn[i]) tj(i);dfs(1);
return 0*printf("%d\n",f[1][0]);
}
Blog来自PaperCloud,未经允许,请勿转载,TKS!