luogu CF438E The Child and Binary Tree

背景：

俯卧撑做得自闭。
$50$多个吧，虽然后面的不标准了，还是感觉到了深深的恶意。

题意：

给出$n$个权值，现在你可以从中选个一个组成一棵二叉树，多组询问求权值和恰好为$s_i$的方案数。

思路

套路题。
设$f_k$表示权值和恰好为$k$的时候的方案数。
$g_i$表示权值为$i$的是否出现过，因为一个权值为$i$的点可以算作一种方案。
$$f_k=\sum _{i=1}^k{g}_i\sum _{j=0}^{k-i}{f}_j{f}_{k-i-j}$$

其中：
$$f_0=1$$

($i$枚举当前的权值，$j$枚举左子树的权值和)
容易发现这是一个卷积的形式，写成生成函数有：
$$F=F^{2}G+1$$

$$F=\frac{1\pm \sqrt{1-4G}}{2G}$$

取负号即可（$F(0)=0$）。
$$F=\frac{1-\sqrt{1-4G}}{2G}$$
考虑到$G(0)=0$，因此不能直接多项式求逆。
上下同时乘上$(1+\sqrt{1-4G})$，得：
$$F=\frac{2}{1+\sqrt{1-4G}}$$
用一个桶搞出$g$，然后多项式开方和求逆即可。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
const LL mod=998244353,G=3,inv_G=332748118;
using namespace std;
	int a[1000010],b[1000010],f[1000010],g[1000010],h[1000010],h1[1000010],h2[1000010],h3[1000010];
	int limit,n,m,l,r[1000010];
int dg(int x,int k)
{
	if(!k) return 1;
	int op=dg(x,k>>1);
	if(k&1) return (LL)op*op%mod*x%mod; else return (LL)op*op%mod;
}
int inv(int x)
{
	return dg(x,mod-2);
}
void init(int n)
{
	limit=1,l=0;
	while(limit<(n<<1))
		limit<<=1,l++;
	for(int i=1;i<limit;i++)
	r[i]=((r[i>>1]>>1)|((i&1)<<(l-1)));
}
void NTT(int *now,int limit,int op)
{
	for(int i=0;i<limit;i++)
		if(i<r[i]) swap(now[i],now[r[i]]);
	for(int mid=1;mid<limit;mid<<=1)
	{
		int wn=dg(op==1?G:inv_G,(mod-1)/(mid<<1));
		for(int j=0;j<limit;j+=(mid<<1))
		{
			int w=1;
			for(int k=0;k<mid;k++,w=((LL)w*wn)%mod)
			{
				int x=now[j+k],y=(LL)w*now[j+k+mid]%mod;
				now[j+k]=(x+y)%mod;
				now[j+k+mid]=(x-y+mod)%mod;
			}
		}
	}
}
void dft(int *f,int n,int limit)
{
	NTT(f,limit,-1);
	int INV=inv(limit);
	for(int i=0;i<n;i++)
		f[i]=(LL)f[i]*INV%mod;
}
void poly_inv(int *f,int *g,int n)
{
	if(n==1)
	{
		g[0]=inv(f[0]);
		return;
	}
	poly_inv(f,g,(n+1)>>1);
	init(n);
	memset(a,0,sizeof(a));
	memset(b,0,sizeof(b));
	for(int i=0;i<n;i++)
		a[i]=f[i],b[i]=g[i];
	NTT(a,limit,1),NTT(b,limit,1);
	for(int i=0;i<limit;i++)
		b[i]=(LL)b[i]*((2ll-(LL)a[i]*b[i]%mod+mod)%mod)%mod;
	dft(b,n,limit);
	for(int i=0;i<n;i++)
		g[i]=b[i];
}
void poly_sqrt(int *f,int *g,int n)
{
	if(n==1)
	{
		g[0]=1;
		return;
	}
	poly_sqrt(f,g,(n+1)>>1);
	init(n);
	memset(a,0,sizeof(a));
	for(int i=0;i<n;i++)
		a[i]=g[i];
	NTT(a,limit,1);
	for(int i=0;i<limit;i++)
		a[i]=(LL)a[i]*a[i]%mod;
	NTT(a,limit,-1);
	int INV=inv(limit);
	memset(h1,0,sizeof(h1));
	memset(h2,0,sizeof(h2));
	memset(h3,0,sizeof(h3));
	for(int i=0;i<n;i++)
		h1[i]=((LL)a[i]*INV%mod+f[i])%mod;
	for(int i=0;i<n;i++)
		h2[i]=2ll*g[i]%mod;
	poly_inv(h2,h3,n);
	NTT(h1,limit,1),NTT(h3,limit,1);
	for(int i=0;i<limit;i++)
		h1[i]=(LL)h1[i]*h3[i]%mod;
	NTT(h1,limit,-1);
	for(int i=0;i<n;i++)
		g[i]=(LL)h1[i]*INV%mod;
}
void INIT(int n)
{
	f[0]=1;
	for(int i=1;i<=n;i++)
		f[i]=(-4ll*f[i]%mod+mod)%mod;
	poly_sqrt(f,g,n);
	g[0]=(g[0]+1)%mod;
	poly_inv(g,h,n);
	for(int i=0;i<=n;i++)
		f[i]=2ll*h[i]%mod;
}
int main()
{
	int x;
	scanf("%d %d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&x);
		f[x]=1;
	}
	INIT(100010);
	for(int i=1;i<=m;i++)
		printf("%d\n",f[i]);
}