BZOJ3684 大朋友和多叉树（多项式相关计算）

设$f(x)$为树的生成函数，即$x^i$的系数为根节点权值为$i$的树的个数。
不难得出$f(x)=\sum_{k\in D}f(x)^k+x$
我们要求这个多项式的第$n$项，由拉格朗日反演可得
$[x^n]f(x)=\frac1n[x^{n-1}](\frac x{g(x)})^n$
其中$[x^n]f(x)$表示$f(x)$的$n$次项系数。
$f(x)$是$g(x)$的复合逆，即$g(f(x))=x$
在本题中，$g(x)=x-\sum_{k\in D}x^k$
我们需要多项式求逆和多项式快速幂。
多项式求逆就不介绍了，多项式快速幂一种朴素的做法是倍增+NTT，复杂度是$O(n\log n\log k)$
有没有更快的做法呢？
观察到$f(x)^n=e^{n\ln(f(x))}$，所以我们只需要快速算$\ln(f(x))$及$e^{f(x)}$即可。
注意$f(x)$的常数项要为1，还好出题人良心保证了这一点。
Part 1:如何算$\ln(f(x))$？
设$g(x)=\ln(f(x))$，那么$g'(x)=\frac{f'(x)}{f(x)}$，所以$g(x)=\int\frac{f'(x)}{f(x)}$，时间复杂度$O(n\log n)$
Part 2:如何算$e^{f(x)}$？
还是考虑倍增，假设我已经求出$g_0(x)=e^{f(x)}(mod\;x^n)$，要求$g(x)=e^{f(x)}(mod\;x^{2n})$
根据泰勒展开，有$$0=h(g(x))=\sum_{i=0}^{\infty}\frac{h^{(i)}(g_0(x))}{i!}(g(x)-g_0(x))^i$$当$i>1$时，上式$mod\;x^{2n}$为$0$
所以$0=h(g_0(x))+h'(g_0(x))(g(x)-g_0(x))\;(mod\;x^{2n})$
即$g(x)=g_0(x)-\frac{h(g_0(x))}{h'(g_0(x))}(mod\;x^{2n})$
其中$h(g(x))=\ln(g(x))-f(x)$
所以$g(x)=g_0(x)-\frac{\ln(g_0(x))-f(x)}{\frac 1{g_0(x)}}=g_0(x)(1-\ln(g_0(x))+f(x))\;(mod\;x^{2n})$
时间复杂度$O(n\log n)$

#include <cstdio>
#include <cstring>
#include <algorithm>
#define pre m=n<<1; for(int i=0;i<m;i++) r[i]=(r[i>>1]>>1)|((i&1)<<l)

typedef long long ll;
const int p=,N=;
int n,m,l,x,nn,f[N],g[N],t1[N],t2[N],t3[N],r[N],ni[N];
ll pw(ll a,int b) {ll r=; for(;b;b>>=,a=a*a%p) if(b&) r=r*a%p; return r;}

void ntt(int *a,int n,int f) {
    for(int i=;i<n;i++) if(r[i]>i) std::swap(a[i],a[r[i]]);
    for(int i=;i<n;i<<=)
    for(int j=,wn=pw(,((p-)/(i*)*f+p-)%(p-));j<n;j+=i<<)
    for(int k=,w=;k<i;k++,w=(ll)w*wn%p) {
        int x=a[j+k],y=(ll)a[j+k+i]*w%p;
        a[j+k]=(x+y)%p,a[j+k+i]=(x-y+p)%p;
    }
    if(!~f) for(int i=;i<n;i++) a[i]=(ll)a[i]*ni[n]%p;
}
void inv(int *f,int *g,int *t,int n,int l) {
    if(n==) {g[]=pw(f[],p-); return;}
    inv(f,g,t,n>>,l-),memcpy(t,f,sizeof(int)*n),memset(t+n,,sizeof(int)*n),pre;ntt(t,m,),ntt(g,m,);
    for(int i=;i<m;i++) g[i]=(ll)g[i]*(-(ll)t[i]*g[i]%p+p)%p;
    ntt(g,m,-),memset(g+n,,sizeof(int)*n);
}
void ln(int *f,int *g,int *t,int n,int l) {
    memset(t,,sizeof(int)*n*),inv(f,t,t1,n,l);
    for(int i=;i+<n;i++) g[i]=(ll)f[i+]*(i+)%p;
    pre;ntt(g,m,),ntt(t,m,);
    for(int i=;i<m;i++) g[i]=(ll)g[i]*t[i]%p;
    ntt(g,m,-);
    for(int i=m-;i;i--) g[i]=(ll)g[i-]*ni[i]%p;
    g[]=,memset(g+n,,sizeof(int)*n);
}
void ex(int *f,int *g,int *t,int n,int l) {
    if(n==) {g[]=; return;}
    ex(f,g,t,n>>,l-),memset(t,,sizeof(int)*n*),ln(g,t,t2,n,l);
    for(int i=;i<n;i++) t[i]=(f[i]-t[i]+p)%p;
    t[]=(t[]+)%p,pre;ntt(t,m,),ntt(g,m,);
    for(int i=;i<m;i++) g[i]=(ll)g[i]*t[i]%p;
    ntt(g,m,-),memset(g+n,,sizeof(int)*n);
}

int main() {
    scanf("%d%d",&n,&m),f[]++,ni[]=,nn=n;
    for(int i=;i<=m;i++) scanf("%d",&x),f[x-]=p-;
    for(m=n,n=,l=;n<=m;n<<=) l++;
    for(int i=;i<=n*;i++) ni[i]=(ll)(p-p/i)*ni[p%i]%p;
    inv(f,g,t1,n,l),memset(f,,sizeof f),ln(g,f,t2,n,l);
    for(int i=;i<n;i++) f[i]=(ll)f[i]*nn%p;
    memset(g,,sizeof g),ex(f,g,t3,n,l),printf("%lld",(ll)g[nn-]*ni[nn]%p);
    return ;
}