[bzoj3625][Codeforces Round #250]小朋友和二叉树 (生成函数)

description

我们的小朋友很喜欢计算机科学,而且尤其喜欢二叉树。
考虑一个含有n个互异正整数的序列c[1],c[2],…,c[n]。如果一棵带点权的有根二叉树满足其所有顶点的权值都在集合{c[1],c[2],…,c[n]}中,我们的小朋友就会将其称作神犇的。并且他认为,一棵带点权的树的权值,是其所有顶点权值的总和。
给出一个整数m,你能对于任意的s(1<=s<=m)计算出权值为s的神犇二叉树的个数吗?请参照样例以更好的理解什么样的两棵二叉树会被视为不同的。
我们只需要知道答案关于998244353(7172^23+1,一个质数)取模后的值。
Input

第一行有2个整数 n,m(1<=n<=10^5; 1<=m<=10^5)。
第二行有n个用空格隔开的互异的整数 c[1],c[2],…,c[n](1<=c[i]<=10^5)。
Output

输出m行,每行有一个整数。第i行应当含有权值恰为i的神犇二叉树的总数。请输出答案关于998244353(=7172^23+1,一个质数)取模后的结果。
Sample Input
样例一:
2 3
1 2
样例二:
3 10
9 4 3
样例三:
5 10
13 10 6 4 15
Sample Output
样例一:
1
3
9
样例二:
0
0
1
1
0
2
4
2
6
15
样例三:
0
0
0
1
0
1
0
2
0
5
Hint

对于第一个样例,有9个权值恰好为3的神犇二叉树:
在这里插入图片描述
Source
VFleaKing & pyx1997 感谢wyl8899提供中文翻译

solution

取模俺就省略不写了
f[i]f[i]f[i]:表示权值和为iii的本质不同的子树个数
g[i]g[i]g[i]:表示子树根节点是否属于∣C∣|C|C
则可以通过枚举左右儿子及自己本身的权值,列出最暴力简单的状态转移方程
fi=∑j=0ig[j]∑k=0i−jfkfi−j−kf_i=\sum_{j=0}^ig[j]\sum_{k=0}^{i-j}f_kf_{i-j-k}fi=j=0ig[j]k=0ijfkfijk
这无非是一个卷积再卷积的形式,于是就跟生成函数沾边了
fff序列的生成函数为F(x)F(x)F(x)ggg序列的生成函数为G(x)G(x)G(x),则有
F=G×F2+1F=G\times F^2+1F=G×F2+1
加1是因为这个子树可能是只有根节点的
F=G×F2+1⇔GF2−F+1=0⇒F=1±1−4G2G⇒F=1−1−4G2GF=G\times F^2+1\Leftrightarrow GF^2-F+1=0\Rightarrow F=\frac{1±\sqrt{1-4G}}{2G}\Rightarrow F=\frac{1-\sqrt{1-4G}}{2G}F=G×F2+1GF2F+1=0F=2G1±14GF=2G114G
套上多项式开根与多项式求逆即可
是一道码农题
在这里插入图片描述

code

#include <cstdio>
#include <iostream>
using namespace std;
#define int long long
#define mod 998244353
#define maxn 300005
int inv2;
int r[maxn], c[maxn], v[maxn], ni[maxn];
int A[maxn], B[maxn], F[maxn], G[maxn];int qkpow( int x, int y ) {int ans = 1;while( y ) {if( y & 1 ) ans = ans * x % mod;x = x * x % mod;y >>= 1;}return ans;
}void NTT( int *h, int opt, int n ) {int len = 1, l = 0;while( len < n ) len <<= 1, l ++;for( int i = 0;i < len;i ++ )r[i] = ( r[i >> 1] >> 1 ) | ( ( i & 1 ) << ( l - 1 ) );for( int i = 0;i < len;i ++ ) if( i < r[i] ) swap( h[i], h[r[i]] );for( int i = 1;i < len;i <<= 1 ) {int omega = qkpow( opt == 1 ? 3 : mod / 3 + 1, ( mod - 1 ) / ( i << 1 ) );for( int j = 0;j < len;j += ( i << 1 ) )for( int k = 0, w = 1;k < i;k ++, w = w * omega % mod ) {int x = h[j + k], y = h[j + k + i] * w % mod;h[j + k] = ( x + y ) % mod;h[j + k + i] = ( x - y + mod ) % mod;}}if( opt == -1 ) {int inv = qkpow( len, mod - 2 );for( int i = 0;i < len;i ++ )h[i] = h[i] * inv % mod;}
}void polyinv( int n, int *f, int *g ) {if( n == 1 ) { g[0] = qkpow( f[0], mod - 2 ); return; }polyinv( ( n + 1 ) >> 1, f, g );for( int i = 0;i < n;i ++ ) A[i] = f[i], B[i] = g[i];NTT( A, 1, n << 1 );NTT( B, 1, n << 1 );for( int i = 0;i < ( n << 1 );i ++ ) A[i] = A[i] * B[i] % mod * B[i] % mod;NTT( A, -1, n << 1 );for( int i = 0;i < n;i ++ )g[i] = ( g[i] + g[i] - A[i] + mod ) % mod;for( int i = 0;i < ( n << 1 );i ++ ) A[i] = B[i] = 0;
}void polysqrt( int n, int *f, int *g ) {if( n == 1 ) { g[0] = 1; return; }polysqrt( ( n + 1 ) >> 1, f, g );for( int i = 0;i < n;i ++ ) F[i] = f[i];polyinv( n, g, G );NTT( F, 1, n << 1 );NTT( G, 1, n << 1 );for( int i = 0;i < ( n << 1 );i ++ ) G[i] = G[i] * F[i] % mod;NTT( G, -1, n << 1 );for( int i = 0;i < n;i ++ )g[i] = ( G[i] + g[i] ) % mod * inv2 % mod;for( int i = 0;i < ( n << 1 );i ++ ) F[i] = G[i] = 0;
}signed main() {inv2 = qkpow( 2, mod - 2 );int n, m;scanf( "%lld %lld", &n, &m );for( int i = 1, x;i <= n;i ++ ) {scanf( "%lld", &x );c[x] ++;}int len = 1;while( len <= m ) len <<= 1;for( int i = 0;i < len;i ++ )c[i] = mod - ( c[i] << 2 );c[0] ++;polysqrt( len, c, v );v[0] = ( v[0] + 1 ) % mod;polyinv( len, v, ni );for( int i = 1;i <= m;i ++ )printf( "%lld\n", ( ni[i] << 1 ) % mod );return 0;
}

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

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

相关文章

11.9模拟:总结

140pts 30100100 qwq 昨天有点被“写完这题就睡”坑到 所以今天状态不太好&#xff1f; 努力给自己找理由.jpg 最大的损失应该就是T1被降智了吧 那其实才是本场最水的一题 看到1e18的数据范围应该刻意往矩乘上想一想的 但我出门就走错了路 利用杨辉三角推了个大式子 从推出那个…

H - Tunnel Warfare HDU - 1540

H - Tunnel Warfare HDU - 1540 题意&#xff1a; n个数顺序排列&#xff0c;左右数相连&#xff0c; 现在有三个操作&#xff1a; 1.摧毁一个位置上的数 2.回复上一次摧毁的数 3.查询包含该位置的最长连续区间长度 题解&#xff1a; 有两个方法&#xff0c;第一个是区间的…

CF802C-Heidi and Library(hard)【费用流】

正题 题目链接:https://www.luogu.com.cn/problem/CF802C 题目大意 你有一个可以放kkk本书的书架&#xff0c;第iii天要求书架上有第aia_iai​种书&#xff0c;购买第iii种书的价格为cic_ici​。 求满足nnn天要求的最小花费。 1≤n,k≤80,1≤ci≤1061\leq n,k\leq 80,1\leq …

2019年1月已到,Java 8 要收费了吗?

根据此前开源中国发起的 Java 版本使用调查&#xff0c;国内的 Java 主力版本仍是 Java 8&#xff0c;有近 70% 的用户表示仍在使用 Java 8。所以对于「Java 8 是否要收费」这个问题&#xff0c;十分有必要阐述清楚&#xff0c;以消除不必要的恐慌。首先要明确一点&#xff0c;…

[NOI2007] 货币兑换 (dp+李超树维护凸包)

description 小Y最近在一家金券交易所工作。该金券交易所只发行交易两种金券&#xff1a;A纪念券&#xff08;以下简称A券&#xff09;和 B纪念券&#xff08;以下简称B券&#xff09;。每个持有金券的顾客都有一个自己的帐户。金券的数目可以是一个实数。每天随着市场的起伏波…

CF1379F1 Chess Strikes Back (easy version)(鸽笼原理、线段树)

解析 很神奇的一道题 关键在于把22的正方形看成一个单位的转化 由于每个22最多有一个国王 因此每个2*2都一定有一个国王 这是本题的关键 个人感觉这个思想很像鸽笼原理 至于后面的线段树就水到渠成了 代码 #include<bits/stdc.h> using namespace std; #define ll l…

[ZJOI2008]树的统计

[ZJOI2008]树的统计 题意&#xff1a; 题解&#xff1a; 树链剖分模板题&#xff0c;好久没打都忘了 代码&#xff1a; #include <algorithm> #include <cstdio> #include <cstring> #define lc o << 1 #define rc o << 1 | 1 const int max…

AT3860-[AGC020F]Arcs on a Circle【dp】

正题 题目链接:https://www.luogu.com.cn/problem/AT3860 题目大意 有一个周长为mmm的圆&#xff0c;nnn条线段&#xff0c;第iii条长度为aia_iai​&#xff0c;将线段贴在圆的随机位置上&#xff0c;求整个圆都被覆盖的概率。 1≤n≤6,1≤m≤501\leq n\leq 6,1\leq m\le 501…

带你学习AOP框架之Aspect.Core[1]

在软件业&#xff0c;AOP为Aspect Oriented Programming的缩写&#xff0c;意为&#xff1a;面向切面编程&#xff0c;通过预编译方式和运行期动态代理实现程序功能的统一维护的一种技术。AOP是OOP的延续&#xff0c;是软件开发中的一个热点&#xff0c;是函数式编程的一种衍生…

「雅礼集训 2017 Day7」事情的相似度(后缀自动机+LCT+树状数组)

description 点击查看题目内容 solution Step1 无脑建SAMSAMSAM 两个前缀的最长公共后缀就是parent−treeparent-treeparent−tree上两点的lcalcalca&#xff0c;定义知显然 Step2 离线询问&#xff0c;按右端点从小到大排序 Step3 每加入一个字母&#xff0c;就将tatata在p…

CF1396B Stoned Game(博弈论)

解析 上午模拟被博弈论虐&#xff0c;下午被黄色的博弈论虐 qwq 首先的一个结论是&#xff1a; 如果有一堆超过总体的一半&#xff0c;那么先手只需要守住这堆就稳赢了 如果没有这样的一堆&#xff0c;可以证明&#xff0c;最后一定可以全部拿完 假设最后有一堆k没拿完&…

主席树

大一寒假的时候学了主席树&#xff0c;当时还录了视频发b站上&#xff0c;过了好久没用都忘了差不多 这篇文章讲的非常详细&#xff0c;我就直接借用过来 静态主席树 总结一下就是&#xff1a; 代码&#xff1a; 对于每一种状态建立一个线段树&#xff0c;为了防止mle&#xf…

P7735-[NOI2021]轻重边【树链剖分,线段树】

前言 之前线上赛就A的题现在才写博客 正题 题目链接:https://www.luogu.com.cn/problem/P7735 题目大意 有nnn个点的一棵树&#xff0c;开始所有边都是轻边&#xff0c;mmm次操作。 把x→yx\rightarrow yx→y路径上所有点连接的重边都变为轻边&#xff0c;然后再把路径上的…

.NET Core实战项目之CMS 第十六章 用户登录及验证码功能实现

前面为了方便我们只是简单实现了基本业务功能的增删改查&#xff0c;但是登录功能还没有实现&#xff0c;而登录又是系统所必须的&#xff0c;得益于 ASP.NET Core的可扩展性因此我们很容易实现我们的登录功能。今天我将带着大家一起来实现下我们的ASP.NET Core2.2开发的CMS系统…

CF508D Tanya and Password(欧拉回路)

解析 之前模拟考过的一道题 把字符串当成前后缀之间的连边即可 注意即使图的度数符合要求&#xff0c;也可能由于不连通而无解&#xff0c;需要再特判一下 #include<bits/stdc.h> using namespace std; #define ll long long #define ull unsigned long long const int…

[HEOI2016/TJOI2016]字符串 (后缀数组+主席树+二分)

description 佳媛姐姐过生日的时候&#xff0c;她的小伙伴从某东上买了一个生日礼物。生日礼物放在一个神奇的箱子中。箱子外边写了一个长为 n 的字符串 s&#xff0c;和 m 个问题。佳媛姐姐必须正确回答这 m 个问题&#xff0c;才能打开箱子拿到礼物&#xff0c;升职加薪&…

[SDOI2014]旅行

[SDOI2014]旅行 题意&#xff1a; n个城市&#xff0c;n-1条边&#xff0c;任意两个城市互通&#xff0c;每个城市有所信奉的宗教和城市评级&#xff0c;有四种指令&#xff1a; 1.将城市x的居民改信为c教 2.将城市x的评级调整为w 3.统计x到y&#xff0c;路上所有的城市的评级…

YbtOJ-染色计划【树链剖分,线段树,tarjan】

正题 题目大意 给出nnn个点的一棵树&#xff0c;每个点有个颜色aia_iai​&#xff0c;你每次可以选择一个颜色全部变成另一个颜色。 求最少多少次操作可以把一种颜色变成一个完整的连通块。 1≤k≤n≤21051\leq k\leq n\leq 2\times 10^51≤k≤n≤2105 解题思路 考虑如果我们…

张高兴的 .NET Core IoT 入门指南:环境配置、Blink、部署

如何在 Raspberry Pi 的 Raspbian 上构建使用 GPIO 引脚的 IoT 程序&#xff1f;你可能会回答使用 C 或 Python 去访问 Raspberry Pi 的引脚。现在&#xff0c;C# 程序员可以使用 .NET Core 在 Raspbian(Linux) 上构建 IoT 应用程序。只需要引入 System.Device.GPIONuGet 包即可…

CF183C:Diverse Permutation(构造)

解析 首先考虑如何构造kn-1的情况 发现令排列为1,n,2,n-1…即可 那么在k不等于n-1时 只需要让前k项与上面的序列相同&#xff0c;后面公差为1即可 代码 #include<bits/stdc.h> using namespace std; #define ll long long #define ull unsigned long long const int N…