Submission #17680444

---

Source Code Expand

#include<bits/stdc++.h>
using namespace std;
#define M 998244353
#define N 200000
typedef long long int LL;
LL fact[301],invfact[301],sum[301],sum2[301];
int ap[N][301],p2[301];
LL mult(LL a, LL b){
	return a*b%M;
}
LL add(LL a, LL b){
	return (a+b+M)%M;
}
LL c(LL n, LL k){
	return mult(fact[n],mult(invfact[k],invfact[n-k]));
}
LL p(LL b, LL x){
	if(x==0) return 1LL;
	LL sub=p(b,x/2);
	sub=mult(sub,sub);
	if(x&1) sub=mult(sub,b);
	return sub;
}
int main(){
	int n,k;
	cin>>n>>k;
	int a[n];
	for(int i=0;i<n;i++) scanf("%d",a+i);
	for(int i=0;i<n;i++) ap[i][0]=1;
	for(int i=0;i<n;i++) for(int j=1;j<=k;j++) ap[i][j]=mult(ap[i][j-1],a[i]);
	fact[0]=invfact[0]=1;
	for(int i=1;i<301;i++) fact[i]=mult(fact[i-1],i),invfact[i]=p(fact[i],M-2);
	p2[0]=1;
	for(int i=1;i<301;i++) p2[i]=mult(2,p2[i-1]);
	for(int i=0;i<n;i++) for(int j=0;j<=k;j++) sum[j]=add(sum[j],ap[i][j]);
	LL tinv=p(2,M-2);
	for(int X=1;X<=k;X++){
		LL ans=0;
		for(int x=0;x<=X;x++){
			ans=add(ans,mult(c(X,x),mult(sum[X-x],sum[x])));
		}
		ans=add(ans,-mult(p2[X],sum[X]));
		ans=mult(ans,tinv);
		cout<<ans<<"\n";
	}
}

Submission Info

Compile Error

./Main.cpp: In function ‘int main()’:
./Main.cpp:28:28: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
   28 |  for(int i=0;i<n;i++) scanf("%d",a+i);
      |                       ~~~~~^~~~~~~~~~

Judge Result