CF438E-The Child and Binary Tree【生成函数】

正题

题目链接:https://www.luogu.com.cn/problem/CF438E


题目大意

每个节点有nnn个权值可以选择,对于1∼m1\sim m1m中的每个数字kkk,求权值和为kkk的二叉树个数。


解题思路

fnf_nfn表示权值和为nnn的方案数,gng_ngn表示nnn这个权值是否可用。
那么我们对于一个nnn的转移,可以枚举根节点的权值,然后再用fff去计算子节点的权值,具体的式子是
fn=∑w=1ngw∑i=0n−wfifn−w−if_n=\sum_{w=1}^ng_w\sum_{i=0}^{n-w}f_if_{n-w-i}fn=w=1ngwi=0nwfifnwi
会发现这个三个数的下标和就是nnn,这其实一个大卷积,设多项式F[x]=fxF[x]=f_xF[x]=fxG[x]=gxG[x]=g_xG[x]=gx。那么根据上面式子就有
F=F2G+1F=F^2G+1F=F2G+1
(加1是因为f0=1f_0=1f0=1,然后可以解出式子)
F=1±1−4G2GF=\frac{1\pm \sqrt{1-4G}}{2G}F=2G1±14G
这里的±\pm±我们取负号,因为取正号时不满足收敛性。

当然我也不知道怎么判正负,但是这题发现F[0]=1F[0]=1F[0]=1但是G[0]=1G[0]=1G[0]=1。又有2GF=1±1−4G2GF=1\pm\sqrt{1-4G}2GF=1±14G,显然有(2GF)[0]=0(2GF)[0]=0(2GF)[0]=0。但是如果取正号,那么(1+1−4G)[0]≥1(1+\sqrt{1-4G})[0]\geq1(1+14G)[0]1,所以显然不可能,只能取负。

然后可以直接上多项式开根和求逆做?发现G[0]=0G[0]=0G[0]=0不能求逆,只好再化一下式子变成
F=21+1−4GF=\frac{2}{1+\sqrt{1-4G}}F=1+14G2

时间复杂度O(nlog⁡n)O(n\log n)O(nlogn)


code

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const ll N=8e5+10,P=998244353,inv2=(P+1)/2;
ll n,m,a[N],b[N];
ll power(ll x,ll b){ll ans=1;while(b){if(b&1)ans=ans*x%P;x=x*x%P;b>>=1;}return ans;
}
namespace Poly{ll n,t1[N],t2[N],t3[N],t4[N],r[N];void GetL(ll l){n=1;while(n<=l)n<<=1;for(ll i=0;i<n;i++)r[i]=(r[i>>1]>>1)|((i&1)?(n>>1):0);return;}void NTT(ll *f,ll op){for(ll i=0;i<n;i++)if(i<r[i])swap(f[i],f[r[i]]);for(ll p=2;p<=n;p<<=1){ll len=p>>1,tmp=power(3,(P-1)/p);if(op==-1)tmp=power(tmp,P-2);for(ll k=0;k<n;k+=p){ll buf=1;for(ll i=k;i<k+len;i++){ll tt=f[i+len]*buf%P;f[i+len]=(f[i]-tt+P)%P;f[i]=(f[i]+tt)%P;buf=buf*tmp%P;}}}if(op==-1){ll invn=power(n,P-2);for(ll i=0;i<n;i++)f[i]=f[i]*invn%P;}return;}void GetInv(ll *f,ll *g,ll m){if(m==1){g[0]=power(f[0],P-2);return;}GetInv(f,g,m>>1);GetL(m);for(ll i=0;i<m;i++)t1[i]=f[i],t2[i]=g[i];for(ll i=m;i<n;i++)t1[i]=t2[i]=0;NTT(t1,1);NTT(t2,1);for(ll i=0;i<n;i++)t1[i]=t1[i]*t2[i]%P*t2[i]%P;NTT(t1,-1);for(ll i=0;i<m;i++)g[i]=(2*g[i]-t1[i]+P)%P;return;}void GetSqrt(ll *f,ll *g,ll m){if(m==1){g[0]=1;return;}GetSqrt(f,g,m>>1);for(ll i=0;i<m;i++)t3[i]=0;GetInv(g,t3,m);GetL(m<<1);for(ll i=0;i<m;i++)t4[i]=f[i];for(ll i=m;i<n;i++)t3[i]=t4[i]=0;NTT(t3,1);NTT(t4,1);for(ll i=0;i<n;i++)t3[i]=t3[i]*t4[i]%P;NTT(t3,-1);for(ll i=0;i<m;i++)g[i]=(g[i]+t3[i])*inv2%P;return;}
}
signed main()
{scanf("%lld%lld",&n,&m);for(ll i=0;i<n;i++){ll x;scanf("%lld",&x);if(x>m)continue;a[x]=P-4;}ll l=1;a[0]++;while(l<=m)l<<=1;Poly::GetSqrt(a,b,l);memset(a,0,sizeof(a));b[0]++;Poly::GetInv(b,a,l);for(ll i=1;i<=m;i++)printf("%lld\n",a[i]*2%P);return 0;
}

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

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

相关文章

纪中A组模拟赛总结(2021.7.13)

成绩 rankrankranknamenamenamescorescorescoreT1T1T1T2T2T2T3T3T3121212lyflyflyf150150150100100100303030202020总结 比赛的时候TJH直接怒言&#xff1a;T1不是网络流吗&#xff01;&#xff01;&#xff01;然后想了想网络流&#xff0c;发现可以用费用流&#xff0c;但只…

牛客网 【每日一题】7月24日题目精讲—小A的柱状图

链接&#xff1a; 文章目录题目描述题解&#xff1a;代码&#xff1a;小A的柱状图时间限制&#xff1a;C/C 1秒&#xff0c;其他语言2秒 空间限制&#xff1a;C/C 262144K&#xff0c;其他语言524288K 64bit IO Format: %lld题目描述 柱状图是有一些宽度相等的矩形下端对齐以后…

在asp.net core2.1中添加中间件以扩展Swashbuckle.AspNetCore3.0支持简单的文档访问权限控制...

Swashbuckle.AspNetCore3.0 介绍一个使用 ASP.NET Core 构建的 API 的 Swagger 工具。直接从您的路由&#xff0c;控制器和模型生成漂亮的 API 文档&#xff0c;包括用于探索和测试操作的 UI。项目主页&#xff1a;https://github.com/domaindrivendev/Swashbuckle.AspNetCore划…

Codeforces Round #669 (Div. 2)

那天晚上和同学打球打的有点晚&#xff0c;结果就鸽了&#xff0c;现在来补一下 A - Ahahahahahahahaha 直接看原数组中0的个数cnt0和1的个数cnt1&#xff0c;谁多留谁即可&#xff0c;注意留1的时候要留偶数个。 #define IO ios::sync_with_stdio(false);cin.tie();cout.ti…

P3235-[HNOI2014]江南乐【整除分块,SG函数】

正题 题目链接:https://www.luogu.com.cn/problem/P3235 题目大意 TTT组游戏&#xff0c;固定给出FFF。每组游戏有nnn个石头&#xff0c;每次操作的人可以选择一个数量不少于FFF的石堆并把它尽量均摊成MMM堆(M>1)(M>1)(M>1)。无法操作的人输&#xff0c;求每组游戏是…

【线段树】二进制(luogu 4428)

正题 luogu 4428 题目大意 给你一个01串&#xff0c;让你进行一下两种操作&#xff1a; 1.将其中一位取反 2.问你某一段中有多少个子串满足有一种排列方案&#xff0c;使得组成的二进制数是3的倍数 解题思路 不难发现&#xff0c;因为2%32&#xff0c;所以2的幂%3的结果按12…

牛客网【每日一题】7月29日题目精讲—Max Power

来源&#xff1a;牛客网&#xff1a; Max Power 时间限制&#xff1a;C/C 1秒&#xff0c;其他语言2秒 空间限制&#xff1a;C/C 32768K&#xff0c;其他语言65536K 64bit IO Format: %lld题目描述 小卤蛋刚把dnf的技能点重新洗了一遍,现在他要重新加点,假设他的技能树一共有…

Swashbuckle.AspNetCore3.0的二次封装与使用

关于 Swashbuckle.AspNetCore3.0一个使用 ASP.NET Core 构建的 API 的 Swagger 工具。直接从您的路由&#xff0c;控制器和模型生成漂亮的 API 文档&#xff0c;包括用于探索和测试操作的 UI。项目主页&#xff1a;https://github.com/domaindrivendev/Swashbuckle.AspNetCore项…

牛客练习赛 69

第一次打牛客直接。。。 y1s1牛客的评测系统真的慢&#xff0c;搞得我不想交 B - 划分 题目链接 首先先对数组a[]逆序贪心可得val(i,j)a1a2⋯aijval(i,j)a_1a_2\dotsa_{ij}val(i,j)a1​a2​⋯aij​ 尝试证明&#xff1a;分析可知我们最终会选择ijijij个数组a[]的数&#xff0…

P3645-[APIO2015]雅加达的摩天楼【bfs,根号分治】

正题 题目链接:https://www.luogu.com.cn/problem/P3645 题目大意 nnn个点&#xff0c;mmm条狗&#xff0c;第iii条狗可以往左或者右跳恰好pip_ipi​步&#xff0c;开始是000号狗&#xff0c;每次跳跃到达一个点可以选择换一条狗&#xff0c;求到111号狗所在点的最短路。 解题…

纪中A组模拟赛总结(2021.7.14)

成绩 rankrankranknamenamenamescorescorescoreT1T1T1T2T2T2T3T3T3393939lyflyflyf232323181818555000总结 比赛的时候T1是另一道题&#xff0c;就是一道二分DP&#xff0c;推了一会就写出来了 然后就在推T2&#xff0c;推了好久只推了一点 考到一半的时候XC过来说T1改题了&…

.NET微服务调查结果

.NET Core就是专门针对模块化的微服务架构而设计, 在2018年国庆时间展开.NET微服务的使用情况&#xff0c;本次调查我们总计收到了来自378个开发者的调查。从落地现状、架构体系、未来趋势等方面对微服务进行了分析。希望能够为传统企业微服务决策、规划和实施提供依据和解决办…

牛客网【每日一题】7月30日题目精讲—Xor Path

来源&#xff1a;牛客网&#xff1a; Xor Path 时间限制&#xff1a;C/C 1秒&#xff0c;其他语言2秒 空间限制&#xff1a;C/C 262144K&#xff0c;其他语言524288K 64bit IO Format: %lld题目描述 给定一棵n个点的树&#xff0c;每个点有权值A i。定义path(i,j)表示i 到j 的…

Codeforces Round #670 (Div. 2)

A - Subset Mex 知道NIM游戏应该都知道Mex。那就直接贪心分组&#xff0c;保证尽可能每组都存在每一个自然数&#xff0c;然后按照Mex定义直接求答案即可 #define IO ios::sync_with_stdio(false);cin.tie();cout.tie(0) #pragma GCC optimize(2) #include<iostream> #…

纪中A组模拟赛总结(2021.7.15)

成绩 rankrankranknamenamenamescorescorescoreT1T1T1T2T2T2T3T3T3T4T4T4212121lyflyflyf113113113000333333303030505050 总结 T1开始想到可以暴力枚举质因数&#xff0c;然后可以跑一个线性同于方程&#xff0c;然而我完全不会扩欧&#xff0c;就乱打了一个费马小&#xff0…

P3703-[SDOI2017]树点涂色【LCT,线段树】

正题 题目链接:https://www.luogu.com.cn/problem/P3703 题目大意 nnn个点的一棵树开始所有点有不同的颜色&#xff0c;mmm次操作 将根节点到xxx节点的路径上染上一种新的颜色询问一条路径的不同颜色个数询问一个节点的子树中的一个xxx使得xxx到根节点的颜色最多。 解题思路 …

牛客练习赛 68

A.牛牛的mex n,q≤105,0≤ai<n且ai互不相同n,q≤10 ^5,0≤a i<n 且 a_i互不相同n,q≤105,0≤ai<n且ai​互不相同 后面两个条件非常重要&#xff0c;通过后面两个条件将问题转化为一个区间内最小未出现的自然数就等于不在这个区间内最小出现的自然数对于区间[l,r][l,r]…

牛客网【每日一题】7月31日题目精讲—兔子的区间密码

来源&#xff1a;牛客网 &#xff1a; 时间限制&#xff1a;C/C 1秒&#xff0c;其他语言2秒 空间限制&#xff1a;C/C 32768K&#xff0c;其他语言65536K 64bit IO Format: %lld文章目录题目描述题解&#xff1a;代码&#xff1a;题目描述 有一只可爱的兔子被困在了密室了&a…

Redis 实战之主从复制、高可用、分布式

简介本节内容基于 CentOS 7.4.1708&#xff0c;Redis 3.2.12 环境实验。Redis 是一个开源的高性能键值对数据库。安装&#xff1a;yum install -y redis特性&#xff1a;高性能 Key-Value 服务器多种数据结构丰富功能缓存(get|set)计数器(incre)消息队列(publish|subcrib)高可用…

纪中A组模拟赛总结(2021.7.16)

成绩 rankrankranknamenamenamescorescorescoreT1T1T1T2T2T2T3T3T3T4T4T4525252lyflyflyf000000000000000 前言&#xff1a;暴零了&#xff0c;太棒了呀(&#xff3e;&#xff0d;&#xff3e;) 总结 开考后再四题里来回看&#xff0c;都没有找到解题方法&#xff08;主要是有…