LOJ#6389. 「THUPC2018」好图计数 / Count【生成函数】

传送门

解题思路：

终于卡过去了……
这题和求无标号有根树个数的思路差不多，可以先看这里，因为下面一些公式演算会省略中间过程。

设大小为 $n$ 的好图数目为$f_n$，其中连通的数目为$g_n$
注意但$n\ge 2$时，不连通的好图和连通的好图一一对应，即$g_n=f_n/2$

考虑生成函数$F(x)=\sum f_ix^i$，为何方便，我们设$f_0=1$。
一个好图是由若干个连通好图拼成，即

$$F(x)=\prod\limits_{k>0}(\sum\limits_{i\ge0}x^{ik})^{g_k}=\prod\limits_{k>0}(1-x^k)^{-g_k}$$
两边取 $In$再求导得
$$\frac{F'(x)}{F(x)}=\sum\limits_{k>0}kg_k\frac{x^{k-1}}{1-x^k}$$
$$F'(x)=F(x)\sum\limits_{k>0}kg_k\frac{x^{k-1}}{1-x^k}$$
再拆回每一项看：
$$[x^n]F(x)=(n+1)f_{n+1}=\sum\limits_{i=0}^nf_i([x^{n-i}]\sum\limits_{k>0}kg_k\frac{x^{k-1}}{1-x^k})$$

注意$\frac{x^{k-1}}{1-x^k}=x^{k-1}+x^{2k-1}+x^{3k-1}+...$，所以$[x^n]\frac{x^{k-1}}{1-x^k}=[k|n+1]$，所以

$$(n+1)f_{n+1}=\sum\limits_{i=0}^nf_i\sum\limits_{k|n+1-i}kg_k$$
注意 $i=0$时要移项，且 $f_0=1,g_i=f_i/2$，所以得：
$$\frac{n+1}{2}f_{n+1}=\sum\limits_{k|n+1,k<n+1}kg_k+\sum\limits_{i=1}^nf_i\sum\limits_{k|n+1-i}kg_k$$

设$s_i=\sum\limits_{k|i}kg_k$就可以$O(n^2)$递推了，卡常可过。
也可以分治FFT做到$O(nlog^2n)$

#include<bits/stdc++.h>
#define ll unsigned long long
using namespace std;
int getint()
{
    int i=0,f=1;char c;
    for(c=getchar();(c!='-')&&(c<'0'||c>'9');c=getchar());
    if(c=='-')c=getchar(),f=-1;
    for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
    return i*f;
}
const int N=100005;
int T,n=23333,mod;
inline void add(ll &x,int y){x=x+y>=mod?x+y-mod:x+y;}
ll f[N],g[N],s[N];
ll Pow(ll x,int y)
{
    ll res=1;
    for(;y;y>>=1,x=x*x%mod)
        if(y&1)res=res*x%mod;
    return res;
}
int main()
{
    //freopen("lx.in","r",stdin);
    T=getint(),mod=getint();
    f[0]=f[1]=1;for(int i=1;i<=n;i++)s[i]=1;
    for(int i=1;i<n;i++)
    {
        __int128 tmp=0;
        for(int j=0;j<=i;j++)tmp+=f[j]*s[i+1-j];tmp%=mod;
        g[i+1]=tmp*Pow(i+1,mod-2)%mod,f[i+1]=g[i+1]*2%mod;
        for(int j=i+1;j<=n;j+=i+1)add(s[j],tmp);
    }
    while(T--)cout<<f[getint()]<<'\n';
    return 0;
}