Description
我们的小朋友很喜欢计算机科学,而且尤其喜欢二叉树。
考虑一个含有n个互异正整数的序列c[1],c[2],...,c[n]。如果一棵带点权的有根二叉树满足其所有顶点的权值都在集合{c[1],c[2],...,c[n]}中,我们的小朋友就会将其称作神犇的。并且他认为,一棵带点权的树的权值,是其所有顶点权值的总和。
给出一个整数m,你能对于任意的s(1<=s<=m)计算出权值为s的神犇二叉树的个数吗?请参照样例以更好的理解什么样的两棵二叉树会被视为不同的。
我们只需要知道答案关于998244353(7*17*2^23+1,一个质数)取模后的值。
Input
第一行有2个整数 n,m(1<=n<=10^5; 1<=m<=10^5)。
第二行有n个用空格隔开的互异的整数 c[1],c[2],...,c[n](1<=c[i]<=10^5)。
Output
输出m行,每行有一个整数。第i行应当含有权值恰为i的神犇二叉树的总数。请输出答案关于998244353(=7*17*2^23+1,一个质数)取模后的结果。
Sample Input
样例一:
2 3
1 2
样例二:
3 10
9 4 3
样例三:
5 10
13 10 6 4 15
Sample Output
样例一:
1
3
9
样例二:
0
0
1
1
0
2
4
2
6
15
样例三:
0
0
0
1
0
1
0
2
0
5
HINT
对于第一个样例,有9个权值恰好为3的神犇二叉树:
不会多项式求逆和多项式开根的先看看这里。
首先设
设的生成函数为
设表示权值为的二叉树的数量。
设的生成函数为:
当时,
当时,
所以我们就可以得到:
解这个关于的一元二次方程得:
检验:
当时,
舍。
类比可得,当时,。
综上,
分式的分子分母同时乘以并化简得:
然后就是多项式开根+多项式求逆啦。
如果有误在评论区吼一声哦!
代码:
#include<cstdio>
#include<cctype>
#include<algorithm>
using namespace std;
const int mod=998244353,g=3,gi=332748118,inv2=499122177;
int n,m,r[400010],v[400010],b[400010],c[4000010],d[400010],e[400010];
int rd(){
int x=0;
char c;
do c=getchar();
while(!isdigit(c));
do{
x=(x<<1)+(x<<3)+(c^48);
c=getchar();
}while(isdigit(c));
return x;
}
int qpow(int x,int n){
int ret=1;
while(n){
if(n&1)
ret=1ll*ret*x%mod;
x=1ll*x*x%mod;
n>>=1;
}
return ret;
}
void NTT(int *a,int limit,int type){
for(int i=0;i<limit;i++)
if(i<r[i])
swap(a[i],a[r[i]]);
for(int mid=1;mid<limit;mid<<=1){
int wn=qpow(type==1?g:gi,(mod-1)/(mid<<1));
for(int i=0;i<limit;i+=(mid<<1)){
int w=1;
for(int j=0;j<mid;j++,w=1ll*w*wn%mod){
int x=a[i+j],y=1ll*w*a[i+j+mid]%mod;
a[i+j]=x+y;
a[i+j+mid]=x-y;
if(a[i+j]>=mod)
a[i+j]-=mod;
if(a[i+j+mid]<0)
a[i+j+mid]+=mod;
}
}
}
if(!~type){
int inv=qpow(limit,mod-2);
for(int i=0;i<limit;i++)
a[i]=1ll*a[i]*inv%mod;
}
return;
}
void inv(int *a,int *b,int n){
if(n==1){
b[0]=qpow(a[0],mod-2);
return;
}
inv(a,b,(n+1)>>1);
int l=0,limit=1;
while(limit<=(n<<1)){
l++;
limit<<=1;
}
for(int i=0;i<limit;i++)
r[i]=(r[i>>1]>>1)|((i&1)<<(l-1));
for(int i=0;i<n;i++)
c[i]=a[i];
for(int i=n;i<limit;i++)
c[i]=0;
NTT(b,limit,1);
NTT(c,limit,1);
for(int i=0;i<limit;i++)
b[i]=(2ll*b[i]-1ll*b[i]*b[i]%mod*c[i]%mod+mod)%mod;
NTT(b,limit,-1);
for(int i=n;i<limit;i++)
b[i]=0;
return;
}
void sqrt(int *a,int *b,int n){
if(n==1){
b[0]=1;
return;
}
sqrt(a,b,(n+1)>>1);
for(int i=0;i<n;i++)
d[i]=0;
inv(b,d,n);
int l=0,limit=1;
while(limit<=(n<<1)){
l++;
limit<<=1;
}
for(int i=0;i<limit;i++)
r[i]=(r[i>>1]>>1)|((i&1)<<(l-1));
for(int i=0;i<n;i++)
c[i]=a[i];
for(int i=n;i<limit;i++)
c[i]=0;
NTT(d,limit,1);
NTT(c,limit,1);
for(int i=0;i<limit;i++)
d[i]=1ll*c[i]*d[i]%mod;
NTT(d,limit,-1);
for(int i=0;i<n;i++)
b[i]=(1ll*(b[i]+d[i])*inv2)%mod;
for(int i=n;i<limit;i++)
b[i]=0;
return;
}
int main(){
n=rd();
m=rd();
for(int i=1;i<=n;i++)
v[rd()]=-4;
v[0]=1;
m++;
sqrt(v,b,m);
b[0]++;
inv(b,e,m);
for(int i=1;i<m;i++)
printf("%d\n",e[i]*2%mod);
return 0;
}