tree

对于 \(n\) 个点带标号的无根森林,计算所有森林的树的个数的 \(k\) 次方,对 \(998244353\) 取模。


自闭,错了一堆关于长度的问题,这里以后一定要注意

比如需要 \(n\) 次多项式的点值,但是却使用了乘法后的 \(2\times n\) 多项式的前 \(n\) 项点值;对 \(2\times n\) 多项式按长度 \(2\times n\) DFT 直接什么的

以及还是容易写错 exp 板子的问题...


由 \(prufer\) 序列,我们知道树的生成函数是 \(A(x)=\sum_{i\ge 0}\frac{i^{i-2}}{i!}x^i\) ,注意这里我们要算的话需要定义 \(0\) 个点的树个数是 \(0\)

设森林个数的 \(k\) 次方的生成函数是 \(B_k(x)\) ,我们有

\[B_k(x)=\sum_{i\ge 0}i^k\frac{A^i(x)}{i!}
\]

我们可以知道 \(B_0=e^A\)

因为 \(k\) 比较小,并且我们知道 \(B_0\) ,所以考虑建立 \(B_k\) 与 \(B_{k-1}\) 之间的递推关系式

因为关键在于 \(i^k\) 与 \(i^{k-1}\) 不一样,所以考虑求导去凑

\[\begin{aligned}
B_{k-1}'(x)&=\sum_{i\ge 0}\frac{i^{k-1}}{i!}(A^i(x))'\\
&=\sum_{i\ge 0}\frac{i^k}{i!}A^{i-1}(x)A'(x)\\
&=\frac{A'(x)}{A(x)}\sum_{i\ge 0}\frac{i^k}{i!}A^i(x)\\
&=\frac{A'(x)}{A(x)}B_k(x)
\end{aligned}
\]

因此

\[B_k(x)=\frac{A(x)B'_{k-1}(x)}{A'(x)}
\]

然后套上多项式板子即可

复杂度 \(O(kn\log n)\)


Code:

#include <cstdio>
#include <cctype>
#include <algorithm>
using std::min;
const int SIZE=1<<21;
char ibuf[SIZE],*iS,*iT;
//#define gc() (iS==iT?(iT=(iS=ibuf)+fread(ibuf,1,SIZE,stdin),iS==iT?EOF:*iS++):*iS++)
#define gc() getchar()
template <class T>
void read(T &x)
{
int f=0;x=0;char c=gc();
while(!isdigit(c)) f|=c=='-',c=gc();
while(isdigit(c)) x=x*10+c-'0',c=gc();
if(f) x=-x;
}
const int N=(1<<17)+10;
const int mod=998244353,G=3,Gi=332748118;
inline int add(int x,int y){return x+y>=mod?x+y-mod:x+y;}
#define mul(x,y) (1ll*(x)*(y)%mod)
int qp(int d,int k){int f=1;while(k){if(k&1)f=mul(f,d);d=mul(d,d),k>>=1;}return f;}
int turn[N],fac[N],inv[N];
void NTT(int *a,int lim,int typ)
{
int L=-1;for(int i=1;i<lim;i<<=1) ++L;
for(int i=0;i<lim;i++)
{
turn[i]=turn[i>>1]>>1|(i&1)<<L;
if(i<turn[i]) std::swap(a[i],a[turn[i]]);
}
for(int le=1;le<lim;le<<=1)
{
int wn=qp(typ?G:Gi,(mod-1)/(le<<1));
for(int i=0;i<lim;i+=le<<1)
{
int w=1;
for(int j=i;j<i+le;j++,w=mul(w,wn))
{
int x=a[j],y=mul(w,a[j+le]);
a[j]=add(x,y),a[j+le]=add(x,mod-y);
}
}
}
if(!typ)
{
int inv=qp(lim,mod-2);
for(int i=0;i<lim;i++) a[i]=mul(a[i],inv);
}
}
int iva[N],ivb[N];
void polyinv(int *a,int *b,int lim)
{
if(lim==1){b[0]=qp(a[0],mod-2);return;}
polyinv(a,b,lim>>1);
for(int i=0;i<lim<<1;i++) iva[i]=ivb[i]=0;
for(int i=0;i<lim;i++) iva[i]=a[i],ivb[i]=b[i];
NTT(iva,lim<<1,1),NTT(ivb,lim<<1,1);
for(int i=0;i<lim<<1;i++) iva[i]=mul(ivb[i],add(2,mod-mul(iva[i],ivb[i])));
NTT(iva,lim<<1,0);
for(int i=0;i<lim;i++) b[i]=iva[i];
}
void polyqd(int *a,int lim)
{
for(int i=0;i<lim-1;i++) a[i]=mul(a[i+1],i+1);
a[lim-1]=0;
}
void polyint(int *a,int lim)
{
for(int i=lim-1;i;i--)
a[i]=mul(a[i-1],mul(fac[i-1],inv[i]));
a[0]=0;
}
int lna[N],lnb[N];
void polyln(int *a,int lim)
{
for(int i=0;i<lim<<1;i++) lna[i]=lnb[i]=0;
for(int i=0;i<lim;i++) lna[i]=a[i];
polyinv(lna,lnb,lim);
polyqd(lna,lim);
NTT(lna,lim<<1,1),NTT(lnb,lim<<1,1);
for(int i=0;i<lim<<1;i++) lna[i]=mul(lna[i],lnb[i]);
NTT(lna,lim<<1,0);
polyint(lna,lim);
for(int i=0;i<lim;i++) a[i]=lna[i];
}
int exa[N],exb[N];
void polyexp(int *a,int *b,int lim)
{
if(lim==1){b[0]=1;return;}
polyexp(a,b,lim>>1);
for(int i=0;i<lim<<1;i++) exa[i]=exb[i]=0;
for(int i=0;i<lim;i++) exa[i]=exb[i]=b[i];
polyln(exb,lim);
for(int i=0;i<lim;i++) exb[i]=add(a[i]+(i==0),mod-exb[i]);
NTT(exa,lim<<1,1),NTT(exb,lim<<1,1);
for(int i=0;i<lim<<1;i++) exa[i]=mul(exa[i],exb[i]);
NTT(exa,lim<<1,0);
for(int i=0;i<lim;i++) b[i]=exa[i];
}
int A[N],B[21][N],C[N],D[N];
int main()
{
int n,k;
read(n),read(k);
int lim=1;
while(lim<=n) lim<<=1;
fac[0]=1;for(int i=1;i<=lim<<1;i++) fac[i]=mul(fac[i-1],i);
inv[lim<<1]=qp(fac[lim<<1],mod-2);
for(int i=(lim<<1)-1;~i;i--) inv[i]=mul(inv[i+1],i+1);
A[0]=C[0]=0;A[1]=C[1]=1;
for(int i=2;i<lim;i++) A[i]=C[i]=mul(qp(i,i-2),inv[i]); polyexp(A,B[0],lim);
polyqd(C,lim);
polyinv(C,D,lim); NTT(A,lim<<1,1),NTT(D,lim<<1,1);
for(int i=0;i<lim<<1;i++) A[i]=mul(A[i],D[i]); NTT(A,lim<<1,0);//warning
for(int i=lim;i<lim<<1;i++) A[i]=0;
NTT(A,lim<<1,1); for(int i=1;i<=k;i++)
{
polyqd(B[i-1],lim);
for(int j=lim;j<lim<<1;j++) B[i-1][j]=0;//warning
NTT(B[i-1],lim<<1,1);
for(int j=0;j<lim<<1;j++) B[i][j]=mul(B[i-1][j],A[j]);
NTT(B[i],lim<<1,0);
}
printf("%lld\n",mul(B[k][n],fac[n]));
return 0;
}

2019.6.24

tree 解题报告的更多相关文章

  1. 【LeetCode】863. All Nodes Distance K in Binary Tree 解题报告(Python)

    [LeetCode]863. All Nodes Distance K in Binary Tree 解题报告(Python) 作者: 负雪明烛 id: fuxuemingzhu 个人博客: http ...

  2. 【LeetCode】297. Serialize and Deserialize Binary Tree 解题报告(Python)

    [LeetCode]297. Serialize and Deserialize Binary Tree 解题报告(Python) 标签: LeetCode 题目地址:https://leetcode ...

  3. 【LeetCode】331. Verify Preorder Serialization of a Binary Tree 解题报告(Python)

    [LeetCode]331. Verify Preorder Serialization of a Binary Tree 解题报告(Python) 标签: LeetCode 题目地址:https:/ ...

  4. 【LeetCode】109. Convert Sorted List to Binary Search Tree 解题报告(Python)

    [LeetCode]109. Convert Sorted List to Binary Search Tree 解题报告(Python) 标签(空格分隔): LeetCode 作者: 负雪明烛 id ...

  5. 【LeetCode】236. Lowest Common Ancestor of a Binary Tree 解题报告(Python)

    作者: 负雪明烛 id: fuxuemingzhu 个人博客: http://fuxuemingzhu.cn/ 目录 题目描述 题目大意 解题方法 日期 题目地址:https://leetcode.c ...

  6. 【LeetCode】99. Recover Binary Search Tree 解题报告(Python)

    [LeetCode]99. Recover Binary Search Tree 解题报告(Python) 标签(空格分隔): LeetCode 题目地址:https://leetcode.com/p ...

  7. 【LeetCode】662. Maximum Width of Binary Tree 解题报告(Python)

    [LeetCode]662. Maximum Width of Binary Tree 解题报告(Python) 标签(空格分隔): LeetCode 题目地址:https://leetcode.co ...

  8. 【LeetCode】623. Add One Row to Tree 解题报告(Python)

    [LeetCode]623. Add One Row to Tree 解题报告(Python) 标签(空格分隔): LeetCode 题目地址:https://leetcode.com/problem ...

  9. POJ 2054 Color a Tree解题报告

    题干 Bob is very interested in the data structure of a tree. A tree is a directed graph in which a spe ...

  10. 【LeetCode】1161. Maximum Level Sum of a Binary Tree 解题报告 (C++)

    作者: 负雪明烛 id: fuxuemingzhu 个人博客:http://fuxuemingzhu.cn/ 目录 题目描述 题目大意 解题方法 BFS 日期 题目地址:https://leetcod ...

随机推荐

  1. JSONP的产生,和ajax的异同!

    先说说JSONP是怎么产生的: 其实网上关于JSONP的讲解有很多,但却千篇一律,而且云里雾里,对于很多刚接触的人来讲理解起来有些困难,着用自己的方式来阐释一下这个问题,看看是否有帮助. 1.一个众所 ...

  2. Vagrant 手册之网络 - 端口转发

    原文地址 Vagrantfile 配置文件中端口转发的网络标识符:forwarded_port,例如: config.vm.network "forwarded_port", gu ...

  3. Minimum Cost 【POJ - 2516】【网络流最小费用最大流】

    题目链接 题意: 有N个商家它们需要货物源,还有M个货物供应商,N个商家需要K种物品,每种物品都有对应的需求量,M个商家每种物品都是对应的存货,然后再是K个N*M的矩阵表示了K个物品从供货商运送到商家 ...

  4. Support Vector Machine(3):Soft Margin 平衡之美

    很多材料上面讲道“引入Soft Margin的原因是因为数据线性不可分”,个人认为有些错误,其实再难以被分解的数据,如果我们用很复杂的弯弯绕曲线去做,还是可以被分解,并且映射到高维空间后认为其线性可分 ...

  5. yum安装时出现No more mirrors to try.

    可能原因:可能是不正常删除造成的 解决方法: yum clean allyum makecacheyum -y update 然后重新安装

  6. TCL自动化之SSH交互式

    目前ssh工具很多,但是能够轻松运用到自动化脚本中,可以轻松适配任何环境,满足ssh交互式登录的tcl工具包很少 下面是个人在tcl自动化过程中比较满意的一款自动化脚本 通过使用管道方式分装plink ...

  7. 【学习总结】java数据结构和算法-第一章-内容介绍和授课方式

    总目录链接 [学习总结]尚硅谷2019java数据结构和算法 github:javaDSA 目录 几个经典算法面试题 算法和数据结构的重要性 几个经典算法面试题 字符串匹配 暴力法:慢 kmp算法:更 ...

  8. 【学习总结】cpu缓存

    参考链接: cpu缓存java性能问题初探 高速缓存 在内存与cpu寄存器之间,还有一块区域叫做cpu高速缓存,即我们常常说的cache. cache分为L1.L2.L3三级缓存,速度递减,离cpu越 ...

  9. 使用easyui框架 设置时间格式

    之前做的一个商城项目,后台系统页面使用Easyui做的,记录一个当时卡住的地方: 首先通过<table>标记创建数据网格(datagrid) <table class="e ...

  10. ES6——解构赋值

    解构赋值: 注意: 1.左右两边结构必须一样 练习1,2,3 2.右边必须是个东西(有值)练习4 3.声明和赋值不能分开(必须在一句话里完成)练习5 /* 练习1: // let arr = [1,2 ...