【UOJ#50】【UR #3】链式反应(分治FFT,动态规划)

题面

UOJ

题解

首先把题目意思捋一捋,大概就是有\(n\)个节点的一棵树,父亲的编号大于儿子。

满足一个点的儿子有\(2+c\)个,其中\(c\in A\),且\(c\)个儿子是叶子,另外\(2\)个存在子树,且两种点的链接的边是不同的,求方案数。

那么就考虑一个暴力\(dp\),设\(f[i]\)表示有\(i\)个节点的树的个数。

那么枚举它两个有子树的子树大小,然后把编号给取出来,得到:

\[f[i]=\frac{1}{2}\sum_{j}\sum_{k} {i-1\choose j}{i-1-j\choose k}f[j]f[k]
\]

其中存在限制,也就是\(i-1-j-k\in A\),要乘\(\frac{1}{2}\)是因为这两个子树选择的时候存在先后关系,所以同一棵子树可能会被计算两次。

这样子我们就可以做\(O(n^3)\)了。

然后把式子给拆一下:

\[{i-1\choose j}{i-1-j\choose k}=\frac{(i-1)!}{j!k!(i-1-j-k)!}
\]

之和\(i,j,k\)相关的可以直接和\(f[i],f[j],f[k]\)丢到一起,剩下的只和\(i/j+k\)相关。

然后预处理\(g[i]=\sum_j \frac{f[j]}{j!}\frac{f[i-j]}{(i-j)!}\)。

那么

\[\frac{2f[i]}{(i-1)!}=\sum_{p=j+k}g[p]*[i-1-p\in A]
\]

这样子就可以做到\(O(n^2)\)了。

进一步发现,\(g\)和\(\frac{2f}{(i-1)!}\)都是卷积的形式,所以可以直接分治\(FFT\)求解做到\(O(nlog^2)\)。

然而这里有细节、、、乱搞会挂。

注意到分治\(FFT\)的过程是\([l,mid]\rightarrow [mid+1,r]\),而\(g\)数组在求解的时候是\(f\)卷上\(f\),这样子会导致乘的数组的范围是\([0,r-l]\),可能存在一部分值没有被计算过。

而这个值具有对称性,所以我们强制\(j<k\)的时候才贡献\(2f[j]*f[k]\)。

也就是如果\(i<l\)的时候,数组里丢进去的是\(2f[i]\),当\(l\le i\le mid\)的时候丢\(f[i]\),否则丢\(0\)进去。

然后两个\(log\)就能过了。

然而这题看起来有更加优秀的做法,但是窝太菜了就不管了。

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
#define MOD 998244353
#define MAX 840000
int fpow(int a,int b){int s=1;while(b){if(b&1)s=1ll*s*a%MOD;a=1ll*a*a%MOD;b>>=1;}return s;}
int r[MAX],W[MAX];
void NTT(int *P,int opt,int len)
{
int N,l=0;for(N=1;N<len;N<<=1)++l;
for(int i=0;i<N;++i)r[i]=(r[i>>1]>>1)|((i&1)<<(l-1));
for(int i=0;i<N;++i)if(i<r[i])swap(P[i],P[r[i]]);
for(int i=1;i<N;i<<=1)
{
int w=fpow(3,(MOD-1)/(i<<1));W[0]=1;
for(int k=1;k<i;++k)W[k]=1ll*W[k-1]*w%MOD;
for(int j=0,p=i<<1;j<N;j+=p)
for(int k=0;k<i;++k)
{
int X=P[j+k],Y=1ll*W[k]*P[i+j+k]%MOD;
P[j+k]=(X+Y)%MOD;P[i+j+k]=(X+MOD-Y)%MOD;
}
}
if(opt==-1)
{
reverse(&P[1],&P[N]);
for(int i=0,inv=fpow(N,MOD-2);i<N;++i)P[i]=1ll*P[i]*inv%MOD;
}
}
int jc[MAX],inv[MAX],jv[MAX];
int f[MAX],g[MAX],A[MAX];
int L[MAX],R[MAX];
void CDQ(int l,int r)
{
if(l==r)
{
if(l==1)f[l]=1;
else f[l]=1ll*f[l]*inv[2]%MOD*jc[l-1]%MOD;
return;
}
int mid=(l+r)>>1,N;
CDQ(l,mid);
for(N=1;N<=mid-l+1+r-l;N<<=1);
for(int i=l;i<=mid;++i)L[i-l]=g[i];
for(int i=0;i<=r-l;++i)R[i]=A[i]*jv[i];
NTT(L,1,N);NTT(R,1,N);
for(int i=0;i<N;++i)L[i]=1ll*L[i]*R[i]%MOD;
NTT(L,-1,N);
for(int i=mid+1;i<=r;++i)f[i]=(f[i]+L[i-1-l])%MOD;
for(int i=0;i<N;++i)L[i]=R[i]=0;
for(int i=l;i<=mid;++i)L[i-l]=1ll*f[i]*jv[i]%MOD;
for(int i=0;i<=r-l;++i)
if(i<l)R[i]=2ll*f[i]*jv[i]%MOD;
else if(i<=mid)R[i]=1ll*f[i]*jv[i]%MOD;
NTT(L,1,N);NTT(R,1,N);
for(int i=0;i<N;++i)L[i]=1ll*L[i]*R[i]%MOD;
NTT(L,-1,N);
for(int i=mid+1;i<=r;++i)g[i]=(g[i]+L[i-l])%MOD;
for(int i=0;i<N;++i)L[i]=R[i]=0; CDQ(mid+1,r);
}
int n;
char ch[MAX];
int main()
{
scanf("%d%s",&n,ch);
for(int i=0;i<n;++i)A[i]=ch[i]-48;
jc[0]=jv[0]=inv[0]=inv[1]=1;
for(int i=2;i<=n;++i)inv[i]=1ll*inv[MOD%i]*(MOD-MOD/i)%MOD;
for(int i=1;i<=n;++i)jc[i]=1ll*jc[i-1]*i%MOD;
for(int i=1;i<=n;++i)jv[i]=1ll*jv[i-1]*inv[i]%MOD;
f[1]=1;
/*
for(int i=2;i<=n;++i)
{
for(int j=0;j<=i;++j)g[i]=(g[i]+1ll*f[j]*jv[j]%MOD*f[i-j]%MOD*jv[i-j]%MOD)%MOD;
for(int j=0;j<i;++j)
if(A[i-1-j])
f[i]=(f[i]+1ll*g[j]*jv[i-1-j])%MOD;
f[i]=1ll*f[i]*inv[2]%MOD*jc[i-1]%MOD;
}
*/
CDQ(1,n);
for(int i=1;i<=n;++i)printf("%d\n",f[i]);
return 0;
}

【UOJ#50】【UR #3】链式反应(分治FFT,动态规划)的更多相关文章

  1. 【LOJ#575】【LNR#2】不等关系(容斥,动态规划,分治FFT)

    [LOJ#575][LNR#2]不等关系(容斥,动态规划,分治FFT) 题面 LOJ 题解 一个暴力\(dp\),设\(f[i][j]\)表示考虑完了前\(i\)个位置,其中最后一个数在前面所有数中排 ...

  2. 【UOJ#76】【UR #6】懒癌(动态规划)

    [UOJ#76][UR #6]懒癌(动态规划) 题面 UOJ 题解 神....神仙题. 先考虑如果是完全图怎么做... 因为是完全图,所以是对称的,所以我们只考虑一个有懒癌的人的心路历程. 如果只有一 ...

  3. 【UOJ#22】【UR #1】外星人(动态规划)

    [UOJ#22][UR #1]外星人(动态规划) 题面 UOJ 题解 一道简单题? 不难发现只有按照从大往小排序的顺序选择的才有意义,否则先选择一个小数再去模一个大数是没有意义的. 设\(f[i][j ...

  4. uoj#401. 【CTSC2018】青蕈领主(分治FFT)

    传送门 话说分治\(FFT\)是个啥子啊--还有题目里那字好像念(蕈xùn) 首先考虑无解的情况:区间相交或者\(L_n\neq n\) 这两个都可以感性理解一下 所以区间之间只会有包含关系,我们把每 ...

  5. UOJ#401. 【CTSC2018】青蕈领主 分治,FFT

    原文链接www.cnblogs.com/zhouzhendong/p/UOJ401.html 题解 首先,对于一个排列,它的连续段一定只有包含关系,没有相交关系. 我们可以据此得到一棵表示连续段的树. ...

  6. 再探快速傅里叶变换(FFT)学习笔记(其三)(循环卷积的Bluestein算法+分治FFT+FFT的优化+任意模数NTT)

    再探快速傅里叶变换(FFT)学习笔记(其三)(循环卷积的Bluestein算法+分治FFT+FFT的优化+任意模数NTT) 目录 再探快速傅里叶变换(FFT)学习笔记(其三)(循环卷积的Blueste ...

  7. [Codeforces 553E]Kyoya and Train(期望DP+Floyd+分治FFT)

    [Codeforces 553E]Kyoya and Train(期望DP+Floyd+分治FFT) 题面 给出一个\(n\)个点\(m\)条边的有向图(可能有环),走每条边需要支付一个价格\(c_i ...

  8. BNUOJ 51279[组队活动 Large](cdq分治+FFT)

    传送门 大意:ACM校队一共有n名队员,从1到n标号,现在n名队员要组成若干支队伍,每支队伍至多有m名队员,求一共有多少种不同的组队方案.两个组队方案被视为不同的,当且仅当存在至少一名队员在两种方案中 ...

  9. hdu 5730 Shell Necklace [分治fft | 多项式求逆]

    hdu 5730 Shell Necklace 题意:求递推式\(f_n = \sum_{i=1}^n a_i f_{n-i}\),模313 多么优秀的模板题 可以用分治fft,也可以多项式求逆 分治 ...

随机推荐

  1. 自己主动化的在程序中显示SVN版本号

    有时候会有这种情况,策划拿着应用过来提一个bug,但我们却不好确定策划的手机上装的应用相应的是那个代码版本号. 为了解决问题.我们希望能在应用上显示出当前应用所相应的代码版本号,即svn版本号. 构想 ...

  2. 创建quickstart报错

    在cmd中创建helloword成功(一开始是mvn package失败,后面又执行了一遍又成功了,应该是网络问题) 然后在eclipse 中创建quickstart,结果pom报错找不到如下包 ma ...

  3. oc12--对象作为参数

    // main.m // 对象作为方法的参数传递 #import <Foundation/Foundation.h> /* 士兵 事物名称: 士兵(Soldier) 属性:姓名(name) ...

  4. 【JSOI 2008】 球形空间产生器

    [题目链接] https://www.lydsy.com/JudgeOnline/problem.php?id=1013 [算法] 高斯消元 [代码] #include<bits/stdc++. ...

  5. AIX&nbsp;常用命令汇总(一)

    命令 内核 如何知道自己在运行 32 位内核还是 64 位内核? 要显示内核启用的是 32 位还是 64 位,可输入以下命令: bootinfo -K 如何知道自己在运行单处理器还是多处理器内核? / ...

  6. AIX的系统备份

    AIX克隆盘即AIX的rootvg的备用替换磁盘,用于保留AIX的原始状态,它可作为软件的升级后出现问题快速回退到原系统的备份手段,也可用于测试两个不同版本的AIX系统.系统可保留两块引导磁盘,而且都 ...

  7. [牛客网练习赛 45 F] Magic Slab 解题报告 (最大权闭合子图)

    interlinkage: https://ac.nowcoder.com/acm/contest/847/F description: solution: 最大权闭合子图; 每个单元格看成一个正权点 ...

  8. NOIP2012 D2 T2 借教室 线段树 OR 二分法

    题目描述: 在大学期间,经常需要租借教室.大到院系举办活动,小到学习小组自习讨论,都需要向学校申请借教室.教室的大小功能不同,借教室人的身份不同,借教室的手续也不一样. 面对海量租借教室的信息,我们自 ...

  9. SQL学习整理

    SQL整理 SQL 对大小写不敏感! 一.对数据的操作 实现功能分类: 1. 增: 1.1 表存在,插入栏位: //插入新的行(按栏位的顺序插入) INSERT INTO Table_1 VALUES ...

  10. Android App退出检测

    app的退出检测是很难的,但是获取app“要退出”的状态就容易多了,退出的瞬间并不是真的退出了,ActivityManager要销毁activity,也需要一些时间和资源的. 先见下面的运行效果:  ...