0%

luoguP6295 有标号 DAG 计数

对每个 \(1\leqslant i\leqslant n\)\(i\) 个点的弱连通有标号有向无环图数目,\(n\leqslant 10^5\)

首先,根据 exp 的组合意义,即有标号对象组成的集合个数,如果我们求出了不一定弱连通的 DAG 数目的 EGF \(G(x)\),设弱连通的 DAG 数目的 EGF 为 \(F(x)\),则 \[ F(x)=\ln(G(x)) \] 显然一张 DAG 一定有至少一个入度为 \(0\) 的点,所以我们尝试枚举度数为 \(0\) 的点的个数,然后我们计算度数为 \(0\) 的点和其它点的连边情况,然后就递归到了子问题,这样你可能会写出以下式子 \[ g_i=\sum\limits_{j=1}^i\binom{i}{j}2^{j(i-j)}g_{i-j} \] 但是写个暴力试一下发现是错的,为什么?因为剩下的点里可能也有入度为 \(0\) 的点,所以会计重。

我们发现,我们实际上不是枚举入度为 \(0\) 的点,而是在钦定入度为 \(0\) 的点,我们设 \(h_{i,j}\)\(i\) 个点的图钦定 \(j\) 个入度为 \(0\) 个点的方案数,\(c_{i,j}\) 为恰好 \(j\) 个入度为 \(0\) 的点的方案数,那么有(这个式子都不知道见了多少遍了): \[ h_{i,j}=\sum\limits_{k=j}^i \binom{k}{j}c_{i,k} \] 由二项式反演: \[ c_{i,j}=\sum\limits_{k=j}^i \binom{k}{j}(-1)^{k-j}h_{i,k} \] 同时类似刚才那个假的式子,我们有 $h_{i,j}=2^{j(i-j)}g_{i-j} $,有 \(g_i=\sum\limits_{j=1}^i c_{i,j}\)

然后我们把能带入的都带入 \[ \begin{aligned} g_n&=\sum\limits_{i=1}^nc_{n,i}\\ &=\sum\limits_{i=1}^n\sum\limits_{j=i}^n\binom{j}{i}(-1)^{j-i}h_{i,j}\\ &=\sum\limits_{i=1}^n\sum\limits_{j=i}^n\binom{j}{i}(-1)^{j-i}\binom{n}{j}2^{j(n-j)}g_{n-j}\\ &=\sum\limits_{j=1}^n\binom{n}{j}2^{j(n-j)}g_{n-j}\sum\limits_{i=1}^{j}\binom{j}{i}(-1)^{i-j}\\ &=\sum\limits_{j=1}^n\binom{n}{j}2^{j(n-j)}g_{n-j}(0-(-1)^j)\\ &=\sum\limits_{j=1}^n\binom{n}{j}(-1)^{j+1}2^{j(n-j)}g_{n-j} \end{aligned} \]

然后注意到有个 \(j(n-j)\),我们采用和 Bluestein 一样的套路,把 \(j(n-j)\) 拆成组合数的和。 \[ j(n-j)=\binom{n}{2}-\binom{j}{2}-\binom{n-j}{2} \]

如果记不住可以考虑组合意义:等价于有两堆石子,各有 \(j\)\(n-j\) 个,然后从两堆石子各取一个的方案数等于任取的方案数减去同时在一堆石子里取的方案数。

这里没有用平方式的那个,因为不一定存在 \(2\) 的二次剩余(虽然 AKrry 证明了如果是 NTT 模数都存在)。

这样就是卷积形式了 \[ \begin{aligned} g_n=\sum\limits_{j=1}^n\binom{n}{j}(-1)^{j+1}2^{j(n-j)}g_{n-j}\\ \frac{g_n}{n!2^{\binom{n}{2}}}=\sum\limits_{j=1}^n\frac{(-1)^{j+1}}{2^{\binom{j}{2}}j!}\frac{g^{n-j}}{2^\binom{n-j}{2}(n-j)!} \end{aligned} \]\[ F(x)=\sum\limits_{i=1}^\infty \frac{(-1)^{i+1}}{2^{\binom{i}{2}}i!}x^i\\ G(x)=\sum\limits_{i=0}^\infty \frac{g_i}{2^{\binom{i}{2}}i!}x^i \]

\(G(x)=F(x)G(x)+1\),故 \(G(x)=\frac{1}{1-F(x)}\)。这样就可以求出 \(g\),然后再做一遍 \(\ln\) 就好了。

代码:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
#include<bits/stdc++.h>
using namespace std;
int const inv2=499122177;
namespace polynomials
{
int const p=998244353,g=3;
int const N=(1<<19)+1;
int w[N],iv[N],r[N],last;
typedef vector<int> vec;
int mod(int x){return x>=p?x-p:x;}
int pw(int x,int y)
{
int res=1;
while(y)
{
if(y&1)res=1ll*res*x%p;
x=1ll*x*x%p;
y>>=1;
}
return res;
}
void init(int n)
{
int lim=1;
while(lim<n)lim<<=1;
iv[1]=1;
for(int i=2;i<=lim;i++)iv[i]=mod(p-1ll*(p/i)*iv[p%i]%p);
for(int i=1;i<lim;i<<=1)
{
int wn=pw(g,(p-1)/(i<<1));
for(int j=0,ww=1;j<i;j++,ww=1ll*ww*wn%p)w[i+j]=ww;
}
}
void ntt(vec &f,int n,int op)
{
f.resize(n);
for(int i=1;i<n;i++)r[i]=(r[i>>1]>>1)|((i&1)?(n>>1):0);
for(int i=1;i<n;i++)if(i<r[i])swap(f[i],f[r[i]]);
for(int i=1;i<n;i<<=1)
for(int j=0;j<n;j+=i<<1)
for(int k=0;k<i;k++)
{
int x=f[j+k],y=1ll*f[i+j+k]*w[i+k]%p;
f[j+k]=mod(x+y);f[i+j+k]=mod(x-y+p);
}
if(op==-1)
{
reverse(&f[1],&f[n]);
for(int i=0;i<n;i++)f[i]=1ll*f[i]*iv[n]%p;
}
}
void getinv(vec f,vec &g,int n)
{
static vec x;
g.resize(n);
if(n==1){g[0]=pw(f[0],p-2);return;}
getinv(f,g,(n+1)>>1);
int lim=1;
while(lim<(n<<1))lim<<=1;
x.resize(lim);
for(int i=0;i<n;i++)x[i]=f[i];
for(int i=n;i<lim;i++)x[i]=0;
g.resize(lim);
ntt(x,lim,1),ntt(g,lim,1);
for(int i=0;i<lim;i++)g[i]=1ll*g[i]*(2-1ll*g[i]*x[i]%p+p)%p;
ntt(g,lim,-1);
g.resize(n);
}
void getln(vec f,vec &g,int n)
{
static vec x;
getinv(f,x,n);
for(int i=0;i<n-1;i++)f[i]=1ll*f[i+1]*(i+1)%p;
f[n-1]=0;
int lim=1;
while(lim<((n<<1)-1))lim<<=1;
ntt(f,lim,1),ntt(x,lim,1);
for(int i=0;i<lim;i++)x[i]=1ll*x[i]*f[i]%p;
ntt(x,lim,-1);
g.resize(n);
g[0]=0;
for(int i=1;i<n;i++)g[i]=1ll*x[i-1]*iv[i]%p;
}
}
using namespace polynomials;
int fac[200005],inv[200005];
vec f,G,ans;
int main()
{
int n;
scanf("%d",&n);
n++;
init(n<<1);
fac[0]=inv[0]=1;
for(int i=1;i<n;i++)fac[i]=1ll*fac[i-1]*i%p;
inv[n-1]=pw(fac[n-1],p-2);
for(int i=n-2;i>=1;i--)inv[i]=1ll*inv[i+1]*(i+1)%p;
f.resize(n);
for(int i=1;i<n;i++)
f[i]=1ll*(((i+1)&1)?(p-1):1)*inv[i]%p*pw(inv2,(1ll*i*(i-1)>>1)%(p-1))%p;
for(int i=0;i<n;i++)f[i]=mod(p-f[i]);
f[0]=mod(f[0]+1);
getinv(f,G,n);
for(int i=0;i<n;i++)G[i]=1ll*G[i]*pw(2,(1ll*i*(i-1)>>1)%(p-1))%p;
getln(G,ans,n);
for(int i=1;i<n;i++)printf("%d\n",1ll*ans[i]*fac[i]%p);
return 0;
}