P5900 无标号无根树计数 题解
不懂为啥都要对原式神秘转化之后再牛顿迭代,直接对原式牛顿迭代即可!完全不用转化!
设无标号有根树的组合类是
设
首先钦定
然后假设已经求出模
考虑如何求
观察到
设最终答案是
则
于是
则
于是
现在我们会算无标号有根树了,设
考虑用
枚举这棵子树的大小,则答案为
发现有两个重心的树,即有一棵大小为
设这棵大小为
则原树分别以两个重心为根形成的有根树完全一致,这样的树不会被重复统计,
反之若原树去掉
所以从大小为
所以有
没有封装多项式类,所以代码比较混乱邪恶,建议谨慎阅读。
#include <cstdio>
#include <cstring>
#include <algorithm>
#define int long long
#define G 3
#define _G 332748118
#define M 998244353
using namespace std;
int n, l, r[6000050], f[6000050], g[6000050], h[6000050], x[6000050], y[6000050], z[6000050], k[6000050], v[6000050];
int P(int x, int y)
{
int q = 1;
for (; y; y >>= 1, x = x * x % M)
if (y & 1)
q = q * x % M;
return q;
}
void F(int *f, int n, int v)
{
for (int i = 0; i < n; ++i)
if (i < r[i])
swap(f[i], f[r[i]]);
for (int L = 2, m; L <= n; L <<= 1)
{
m = L >> 1;
int W = P(v == 1 ? G : _G, (M - 1) / L);
for (int l = 0, r = L - 1; r <= n; l += L, r += L)
{
int o = 1;
for (int p = l; p < l + m; ++p)
{
int x = f[p], y = f[p + m];
f[p] = (x + o * y) % M, f[p + m] = (x + M - o * y % M) % M;
o = o * W % M;
}
}
}
}
void I(int *f, int *g, int n)
{
memset(g, 0, n << 4);
memset(x, 0, n << 4);
g[0] = P(f[0], M - 2);
int L;
for (L = 4;; L <<= 1)
{
memcpy(x, f, L << 2);
memcpy(y, g, L << 3);
l = __lg(L);
for (int i = 0; i < L; ++i)
r[i] = r[i >> 1] >> 1 | (i & 1) << l - 1;
F(x, L, 1);
F(y, L, 1);
for (int i = 0; i < L; ++i)
x[i] = x[i] * y[i] % M;
F(x, L, -1);
int _ = P(L, M - 2);
for (int i = 0; i < L; ++i)
x[i] = (M - x[i] * _ % M) % M;
x[0] = (x[0] + 2) % M;
memset(x + (L >> 1), 0, L << 2);
F(g, L, 1);
F(x, L, 1);
for (int i = 0; i < L; ++i)
g[i] = g[i] * x[i] % M;
F(g, L, -1);
for (int i = 0; i < L; ++i)
g[i] = g[i] * _ % M;
if (L >> 1 >= n)
break;
}
memset(g + n, 0, L - n << 3);
}
void LN(int *f, int *g, int n)
{
memset(h, 0, n << 4);
for (int i = 0; i < n - 1; ++i)
h[i] = (i + 1) * f[i + 1] % M;
I(f, g, n);
int L = 1;
while (L >> 1 < n)
L <<= 1;
l = __lg(L);
for (int i = 0; i < L; ++i)
r[i] = r[i >> 1] >> 1 | (i & 1) << l - 1;
F(g, L, 1);
F(h, L, 1);
for (int i = 0; i < L; ++i)
h[i] = g[i] * h[i] % M;
F(h, L, -1);
int _ = P(L, M - 2);
for (int i = 0; i < L; ++i)
h[i] = h[i] * _ % M;
g[0] = 0;
for (int i = 1; i < n; ++i)
g[i] = h[i - 1] * P(i, M - 2) % M;
memset(g + n, 0, L - n << 3);
}
void EXP(int *f, int *g, int n)
{
memset(g, 0, n << 4);
g[0] = 1;
int L;
for (L = 4;; L <<= 1)
{
LN(g, z, L >> 1);
for (int i = 0; i < L >> 1; ++i)
z[i] = (f[i] + M - z[i]) % M;
z[0] = (z[0] + 1) % M;
l = __lg(L);
for (int i = 0; i < L; ++i)
r[i] = r[i >> 1] >> 1 | (i & 1) << l - 1;
F(g, L, 1);
F(z, L, 1);
for (int i = 0; i < L; ++i)
g[i] = g[i] * z[i] % M;
F(g, L, -1);
int _ = P(L, M - 2);
for (int i = 0; i < L; ++i)
g[i] = g[i] * _ % M;
memset(g + (L >> 1), 0, L << 2);
if (L >> 1 >= n)
break;
}
memset(g + n, 0, L - n << 3);
}
signed main()
{
v[1] = 1;
for (int i = 2; i <= 6e6; ++i)
v[i] = (M - M / i) * v[M % i] % M;
scanf("%lld", &n), ++n;
for (int L = 4;; L <<= 1)
{
memset(g, 0, L << 2);
for (int i = 1; i < L >> 2; ++i)
for (int j = i; j < L >> 1; j += i)
g[j] = (g[j] + f[i] * i % M * v[j]) % M;
EXP(g, k, L >> 1);
for (int i = (L >> 1) - 1; i >= 1; --i)
k[i] = (M - k[i - 1]) % M;
k[0] = 1;
I(k, g, L >> 1);
k[0] = 0;
for (int i = 1; i < L >> 2; ++i)
k[i] = (k[i] + f[i]) % M;
F(g, L, 1);
F(k, L, 1);
for (int i = 0; i < L; ++i)
g[i] = g[i] * k[i] % M;
F(g, L, -1);
int _ = P(L, M - 2);
for (int i = 0; i < L; ++i)
g[i] = g[i] * _ % M;
for (int i = 0; i < L >> 1; ++i)
f[i] = (f[i] + M - g[i]) % M;
if (L >> 1 >= n)
break;
}
--n;
int q = f[n];
for (int i = (n >> 1) + 1; i < n; ++i)
q = (q + M - f[i] * f[n - i] % M) % M;
if (!(n & 1))
{
int u = f[n >> 1];
q = (q + M - u * (u - 1) % M * (M + 1 >> 1) % M) % M;
}
printf("%lld", q);
return 0;
}
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 阿里最新开源QwQ-32B,效果媲美deepseek-r1满血版,部署成本又又又降低了!
· 开源Multi-agent AI智能体框架aevatar.ai,欢迎大家贡献代码
· Manus重磅发布:全球首款通用AI代理技术深度解析与实战指南
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· AI技术革命,工作效率10个最佳AI工具