E. The Child and Binary Tree(生成函数 + 多项式)

E. The Child and Binary Tree

不难写出一个递推式fn=∑i=1ngi∑j=0n−ifjfn−i−jf_n = \sum\limits_{i = 1} ^{n}g_i \sum\limits_{j = 0} ^{n - i}f_jf_{n - i - j}fn=i=1ngij=0nifjfnij,其中gig_igi表示ccc中有没有iii这个数。

F(x)F(x)F(x)fff的生成函数,G(x)G(x)G(x)ggg的生成函数,则有F(x)=G(x)F(x)F(x)+1F(x) = G(x)F(x)F(x) + 1F(x)=G(x)F(x)F(x)+1

因为做卷积[x0]G[x ^ 0]G[x0]G是为000的,所以会没有常数项,得在右边加111

解这个方程有F(x)=1±1−4G(x)2G(x)=21±1−4G(x)F(x) = \frac{1 \pm \sqrt{1 - 4G(x)}}{2G(x)} = \frac{2}{1 \pm \sqrt{1 - 4G(x)}}F(x)=2G(x)1±14G(x)=1±14G(x)2,取x=0x = 0x=011±1\frac{1}{1 \pm 1}1±11,所以负号不可取,F(x)=21+1−4G(x)F(x) = \frac{2}{1 + \sqrt{1 - 4G(x)}}F(x)=1+14G(x)2

所以只要多项式开根,多项式求逆就可以得到答案了。

#include <bits/stdc++.h>using namespace std;const int mod = 998244353, inv2 = mod + 1 >> 1;namespace Quadratic_residue {struct Complex {int r, i;Complex(int _r = 0, int _i = 0) : r(_r), i(_i) {}};int I2;Complex operator * (const Complex &a, Complex &b) {return Complex((1ll * a.r * b.r % mod  + 1ll * a.i * b.i % mod * I2 % mod) % mod, (1ll * a.r * b.i % mod + 1ll * a.i * b.r % mod) % mod);}Complex quick_pow(Complex a, int n) {Complex ans = Complex(1, 0);while (n) {if (n & 1) {ans = ans * a;}a = a * a;n >>= 1;}return ans;}int get_residue(int n) {mt19937 e(233);if (n == 0) {return 0;}if(quick_pow(n, (mod - 1) >> 1).r == mod - 1) {return -1;}uniform_int_distribution<int> r(0, mod - 1);int a = r(e);while(quick_pow((1ll * a * a % mod - n + mod) % mod, (mod - 1) >> 1).r == 1) {a = r(e);}I2 = (1ll * a * a % mod - n + mod) % mod;int x = quick_pow(Complex(a, 1), (mod + 1) >> 1).r, y = mod - x;if(x > y) swap(x, y);return x;}
}const int N = 1e6 + 10;int r[N], inv[N], b[N], c[N], d[N], e[N], t[N];int quick_pow(int a, int n) {int ans = 1;while (n) {if (n & 1) {ans = 1ll * a * ans % mod;}a = 1ll * a * a % mod;n >>= 1;}return ans;
}void get_r(int lim) {for (int i = 0; i < lim; i++) {r[i] = (i & 1) * (lim >> 1) + (r[i >> 1] >> 1);}
}void get_inv(int n) {inv[1] = 1;for (int i = 2; i <= n; i++) {inv[i] = 1ll * (mod - mod / i) * inv[mod % i] % mod;}
}void NTT(int *f, int lim, int rev) {for (int i = 0; i < lim; i++) {if (i < r[i]) {swap(f[i], f[r[i]]);}}for (int mid = 1; mid < lim; mid <<= 1) {int wn = quick_pow(3, (mod - 1) / (mid << 1));for (int len = mid << 1, cur = 0; cur < lim; cur += len) {int w = 1;for (int k = 0; k < mid; k++, w = 1ll * w * wn % mod) {int x = f[cur + k], y = 1ll * w * f[cur + mid + k] % mod;f[cur + k] = (x + y) % mod, f[cur + mid + k] = (x - y + mod) % mod;}}}if (rev == -1) {int inv = quick_pow(lim, mod - 2);reverse(f + 1, f + lim);for (int i = 0; i < lim; i++) {f[i] = 1ll * f[i] * inv % mod;}}
}void polyinv(int *f, int *g, int n) {if (n == 1) {g[0] = quick_pow(f[0], mod - 2);return ;}polyinv(f, g, n + 1 >> 1);for (int i = 0; i < n; i++) {t[i] = f[i];}int lim = 1;while (lim < 2 * n) {lim <<= 1;}get_r(lim);NTT(t, lim, 1);NTT(g, lim, 1);for (int i = 0; i < lim; i++) {int cur = (2 - 1ll * g[i] * t[i] % mod + mod) % mod;g[i] = 1ll * g[i] * cur % mod;t[i] = 0;}NTT(g, lim, -1);for (int i = n; i < lim; i++) {g[i] = 0;}
}void polysqrt(int *f, int *g, int n) {if (n == 1) {g[0] = Quadratic_residue::get_residue(f[0]);return ;}polysqrt(f, g, n + 1 >> 1);polyinv(g, b, n);int lim = 1;while (lim < 2 * n) {lim <<= 1;}get_r(lim);for (int i = 0; i < n; i++) {t[i] = f[i];}NTT(g, lim, 1);NTT(b, lim, 1);NTT(t, lim, 1);for (int i = 0; i < lim; i++) {g[i] = (1ll * inv2 * g[i] % mod + 1ll * inv2 * b[i] % mod * t[i] % mod) % mod;b[i] = t[i] = 0;}NTT(g, lim, -1);for (int i = n; i < lim; i++) {g[i] = 0;}
}void derivative(int *a, int *b, int n) {for (int i = 0; i < n; i++) {b[i] = 1ll * a[i + 1] * (i + 1) % mod;}
}void integrate(int *a, int n) {for (int i = n - 1; i >= 1; i--) {a[i] = 1ll * a[i - 1] * inv[i] % mod;}a[0] = 0;
}void polyln(int *f, int *g, int n) {polyinv(f, b, n);derivative(f, g, n);int lim = 1;while (lim < 2 * n) {lim <<= 1;}get_r(lim);NTT(g, lim, 1);NTT(b, lim, 1);for (int i = 0; i < lim; i++) {g[i] = 1ll * g[i] * b[i] % mod;b[i] = 0;}NTT(g, lim, -1);for (int i = n; i < lim; i++) {g[i] = 0;}integrate(g, n);
}void polyexp(int *f, int *g, int n) {if (n == 1) {g[0] = 1;return ;}polyexp(f, g, n + 1 >> 1);int lim = 1;while (lim < 2 * n) {lim <<= 1;}polyln(g, d, n);for (int i = 0; i < n; i++) {t[i] = (f[i] - d[i] + mod) % mod;}t[0] = (t[0] + 1) % mod;get_r(lim);NTT(g, lim, 1);NTT(t, lim, 1);for (int i = 0; i < lim; i++) {g[i] = 1ll * g[i] * t[i] % mod;t[i] = d[i] =  0;}NTT(g, lim, -1);for (int i = n; i < lim; i++) {g[i] = 0;}
}/*b存放多项式逆,c存放多项式开根,d存放多项式对数ln,e存放多项式指数exp,t作为中间转移数组,如果要用到polyinv,得提前调用get_inv(n)先预先得到我们想要得到的逆元范围。
*/int a[N], n, m;int main() {// freopen("in.txt", "r", stdin);// freopen("out.txt", "w", stdout);// ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);scanf("%d %d", &n, &m);for (int i = 1, x; i <= n; i++) {scanf("%d", &x);a[x]++;}for (int i = 0; i <= 100000; i++) {a[i] = (mod - 4 * a[i]) % mod;}a[0] = 1;get_inv(200010);polysqrt(a, c, 200005);for (int i = 0; i < 200005; i++) {a[i] = c[i];c[i] = 0;}a[0] = (a[0] + 1) % mod;polyinv(a, b, 200005);for (int i = 1; i <= m; i++) {printf("%lld\n", 1ll * b[i] * 2 % mod);}return 0;
}

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/news/313943.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

依赖倒置原则(DIP)、控制反转(IoC)、依赖注入(DI)(C#)

象的控制权交由配置文件控制&#xff0c;然后根据配置文件中的信息&#xff08;程序集类型&#xff09;&#xff0c;通过反射来获取对象&#xff0c;而不是直接new对象&#xff0c;这也是控制反转的一种体现。IoC容器会连接程序中的所有模块&#xff0c;模块将所需对象的控制权…

汉明距离

题目描述 两个整数之间的汉明距离指的是这两个数字对应二进制位不同的位置的数目。 给出两个整数 x 和 y&#xff0c;计算它们之间的汉明距离。 注意&#xff1a; 0 ≤ x, y < 231. 示例: 输入: x 1, y 4输出: 2解释: 1 (0 0 0 1) 4 (0 1 0 0)↑ ↑上面的箭头指…

P3978 [TJOI2015]概率论(生成函数)

P3978 [TJOI2015]概率论 设fif_ifi​表示节点数为iii的二叉树有多少&#xff0c;gig_igi​表示节点数为iii的二叉树有多少叶子节点。 fn∑i0n−1fifn−1−if_n \sum\limits_{i 0} ^{n - 1}f_if_{n - 1 - i}fn​i0∑n−1​fi​fn−1−i​&#xff0c;f01f_0 1f0​1。 对于g…

四种为HttpClient添加默认请求报头的解决方案

HttpClient在Web调用中具有广泛的应用&#xff0c;而为它添加默认请求头是我们经常遇到的需求&#xff0c;本文介绍4种为HttpClient添加默认请求头的方式。直接在创建的HttpClient对象的DefaultRequestHeaders集合中添加报头。class Program{static Task Main()> SendAsync…

Java位运算

java中位运算^,&,<<,>>,<<<,>>>总结 1.^(亦或运算) &#xff0c;针对二进制&#xff0c;相同的为0&#xff0c;不同的为1 public static void main(String[] args) {System.out.println("2^3运算的结果是 :"(2^3));//打印的结果是…

#3456. 城市规划(生成函数,多项式求逆)

#3456. 城市规划 设fnf_nfn​为nnn个点的的点的简单无向连通图数目&#xff0c;gng_ngn​为nnn个点的简单无向图个数&#xff08;不要求联通&#xff09;。 对于gng_ngn​显然有gn2n(n−1)2g_n 2 ^{\frac{n(n - 1)}{2}}gn​22n(n−1)​&#xff0c;共有n(n1)2\frac{n(n 1)}…

【盛派内部分享资料】本期主题:使用JavaScript HTML CSS构建跨平台桌面应用

先上录制视频友情提示&#xff1a;如果公众号内视频无法显示高清视频&#xff0c;您也可以在小程序内观看高清视频&#xff0c;点击下方按钮观看&#xff1a;【盛派内部分享资料】本期主题&#xff1a;使用JavaScript HTML CSS构建跨平台桌面应用本次活动由盛派技术人员伏允坤主…

负载均衡及负载均衡器

负载均衡在分布式系统中&#xff0c;负载均衡&#xff08;load balancing&#xff09;是一种有效的将网络请求分配到多个服务器的过程。通过将负载进行负载均衡&#xff0c;可以有效地改进系统响应时间&#xff0c;提高系统的可用性。随着系统变的愈发复杂&#xff0c;用户增多…

Sequence II (HDU 5919)(主席树)

Sequence II 题目大意是有mmm次询问&#xff0c;每次询问一段区间[l,r][l, r][l,r]&#xff0c;从左到右&#xff0c;如果这个数是在这个区间第一次出现&#xff0c;则记录下其下标&#xff0c; 我们会得到一个新的数组&#xff0c;要求这个数组的中位数是什么。 考虑使用主…

VS Code 摸鱼插件开发小记

插件地址&#xff1a;https://marketplace.visualstudio.com/items?itemNamebugbreeder.vscode-readhub原文链接&#xff1a;https://github.com/alex-yh99/vscode-readhub/blob/master/docs/develop-note.md一、关于插件在 VSCode 中快捷查阅科技动态、开发者资讯等 Readhub …

帕斯卡三角形杨辉三角

给定一个非负整数 numRows&#xff0c;生成杨辉三角的前 numRows 行。 在杨辉三角中&#xff0c;每个数是它左上方和右上方的数的和。 示例: 输入: 5 输出: [[1],[1,1],[1,2,1],[1,3,3,1],[1,4,6,4,1] ]解法 思路1&#xff1a;循环遍历 public List<List<Integer>…

ICPC 南昌现场赛 K:Tree(dsu on tree + 动态开点线段树)

Tree 让我们找满足一下五个条件的(x,y(x, y(x,y)点对有多少&#xff1a; x≠yx \neq yx​yxxx不是yyy的祖先yyy不是xxx的祖先dis(x,y)≤kdis(x, y)\leq kdis(x,y)≤kzzz是x,yx, yx,y的最近公共祖先&#xff0c;valuexvaluey2valuezvalue_x value_y 2value_zvaluex​valuey…

缺失数字

题目描述 给定一个包含 0, 1, 2, …, n 中 n 个数的序列&#xff0c;找出 0 … n 中没有出现在序列中的那个数。 示例 1: 输入: [3,0,1] 输出: 2示例 2: 输入: [9,6,4,2,3,5,7,0,1] 输出: 8解法 思路1&#xff1a;使用额外数组存储0~n&#xff0c;遍历数组&#xff0c;缺失…

9月数据库排行:Microsoft SQL Server分数罕见下滑

DB-Engines 数据库流行度排行榜 9 月更新已发布&#xff0c;排名前二十如下&#xff1a;明显能看出来&#xff0c;整体排名和上个月的保持一致。而上个月出现分数下跌的 PostgreSQL 和 MongoDB 终于“回归正道”&#xff0c;分数有了稍微的上升。与上个月相比&#xff0c;Postg…

P4175 [CTSC2008]网络管理(树状数组套主席树)

P4175 [CTSC2008]网络管理 每次询问两点间的第kkk大&#xff0c;考虑建立nnn棵主席树&#xff0c;每棵主席树记录的是从根节点到当前节点的状态&#xff0c; 由于主席树是一种类似前缀和的数据结构&#xff0c;可以使用差分的方式得到(u,v)(u, v)(u,v)点对之间的主席树为uv−…

Exceptionless使用介绍

一.Exceptionless介绍Exceptionless专注于.net平台提供实时错误和日志报告。主要包括&#xff1a;错误通知、智能分组异常、详细错误报告堆栈跟踪、支持离线、UI查看重要错误和确定优先级、仪表板上的统计信息和趋势、对异常标记为已修复&#xff0c;监视回归、将事件标记为关键…

BIO、NIO 和 AIO

BIO、NIO 和 AIO的区别 https://www.cnblogs.com/barrywxx/p/8430790.html 三种 IO 的用法与原理 netty http://ifeve.com/netty5-user-guide/

#279. [SYZOI Round1] 滑稽♂树(树状数组套主席树)

#279. [SYZOI Round1] 滑稽♂树 子树上的问题&#xff0c;考虑dfsdfsdfs序&#xff0c;第kkk大&#xff0c;可以用主席树嘛&#xff0c;支持修改&#xff0c;那就树状数组上套主席树&#xff0c;参考P4175 [CTSC2008]网络管理&#xff08;树状数组套主席树&#xff09; #incl…

Net Core DocXCore 实现word模板导出

实际工作中&#xff0c;往往有这样的需求&#xff0c;需要导出word&#xff0c;还有各种各样的样式&#xff0c;于是有了word模板导出。实现以下几个需求&#xff1a;1、表单导出2、表格导出3、表单表格混合导出4、实际用例测试解决方案&#xff1a;实现是基于NET Core 2.1 &am…

主席树有关的一些题目(持续更新)

主席树 模板 P3919 【模板】可持久化线段树 1&#xff08;可持久化数组&#xff09; #include <bits/stdc.h>using namespace std;const int N 1e6 10;int root[N], n, m;int ls[N * 25], rs[N * 25], value[N * 25], tot;void build(int &rt, int l, int r) {r…