#590148 (C++14) No.1307 Rotate and Accumulate

• 問題: No.1307 Rotate and Accumulate
• ユーザー: kotatsugame
• 提出日時: 2020-12-07 12:35:19
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 197 ms / 5,000 ms
• コード長: 1,817 bytes
• コンパイル時間: 1,013 ms
• コンパイル使用メモリ: 76,376 KB
• 実行使用メモリ: 9,044 KB
• 最終ジャッジ日時: 2023-10-17 16:28:33
• 合計ジャッジ時間: 4,512 ms
• ジャッジサーバーID （参考情報）: judge13 / judge11

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 4,348 KB
testcase_01	AC	2 ms; 4,348 KB
testcase_02	AC	1 ms; 4,348 KB
testcase_03	AC	1 ms; 4,348 KB
testcase_04	AC	2 ms; 4,348 KB
testcase_05	AC	2 ms; 4,348 KB
testcase_06	AC	2 ms; 4,348 KB
testcase_07	AC	2 ms; 4,348 KB
testcase_08	AC	142 ms; 8,548 KB
testcase_09	AC	147 ms; 8,604 KB
testcase_10	AC	110 ms; 6,144 KB
testcase_11	AC	80 ms; 6,404 KB
testcase_12	AC	107 ms; 6,176 KB
testcase_13	AC	29 ms; 4,348 KB
testcase_14	AC	54 ms; 4,748 KB
testcase_15	AC	192 ms; 9,044 KB
testcase_16	AC	193 ms; 9,044 KB
testcase_17	AC	193 ms; 9,044 KB
testcase_18	AC	171 ms; 8,660 KB
testcase_19	AC	196 ms; 9,044 KB
testcase_20	AC	197 ms; 8,656 KB
testcase_21	AC	2 ms; 4,348 KB

コンパイルメッセージ

a.cpp:6:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]

ソースコード

#line 1 "a.cpp"
#include<iostream>
using namespace std;
#line 1 "/home/kotatsugame/library/math/NTT.cpp"
//998244353,3
//469762049,3
//167772161,3
#include<vector>
template<int mod,int proot>
struct NTT{
	vector<long long>pp,invpp;//memoize proot^(mod-1>>i) and inv
	long long power(long long a,int b)
	{
		long long ret=1;
		while(b)
		{
			if(b&1)ret=ret*a%mod;
			a=a*a%mod;
			b>>=1;
		}
		return ret;
	}
	void dft(vector<int>&A,bool sign,int id)
	{
		if(id==0)return;
		int N=1<<id-1;
		vector<int>F(N),G(N);
		for(int i=0;i<N;i++)
		{
			F[i]=A[i<<1];
			G[i]=A[i<<1|1];
		}
		dft(F,sign,id-1);
		dft(G,sign,id-1);
		long long z=(sign?invpp:pp)[id],p=1;
		for(int i=0;i<N;i++)
		{
			A[i]=(F[i]+p*G[i])%mod;
			A[i+N]=(F[i]-p*G[i])%mod;
			if(A[i+N]<0)A[i+N]+=mod;
			(p*=z)%=mod;
		}
	}
	vector<int>multiply(vector<int>A,vector<int>B)
	{
		if(A.empty()||B.empty())
		{
			return(vector<int>){};
		}
		int N=1,sz=0;
		vector<int>ret(A.size()+B.size()-1);
		while(N<ret.size())N<<=1,sz+=1;
		pp.resize(sz+1);
		invpp.resize(sz+1);
		pp[sz]=power(proot,mod-1>>sz);
		invpp[sz]=power(pp[sz],mod-2);
		for(int i=sz-1;i>0;i-=1)
		{
			pp[i]=pp[i+1]*pp[i+1]%mod;
			invpp[i]=invpp[i+1]*invpp[i+1]%mod;
		}
		A.resize(N);
		B.resize(N);
		dft(A,false,sz);
		dft(B,false,sz);
		for(int i=0;i<N;i++)A[i]=(long long)A[i]*B[i]%mod;
		dft(A,true,sz);
		long long invN=power(N,mod-2);
		for(int i=0;i<ret.size();i++)ret[i]=invN*A[i]%mod;
		return ret;
	}
};
#line 4 "a.cpp"
int N,Q;
int A[1<<17],B[1<<17];
main()
{
	cin>>N>>Q;
	for(int i=0;i<N;i++)cin>>A[i];
	for(int i=0;i<Q;i++)
	{
		int r;cin>>r;B[N-1-r]++;
	}
	NTT<998244353,3>P;
	vector<int>ans=P.multiply(vector<int>(A,A+N),vector<int>(B,B+N));
	for(int i=N;i<ans.size();i++)ans[i-N]+=ans[i];
	for(int i=-1;i<N-1;i++)cout<<ans[(i+N)%N]%998244353<<(i==N-2?"\n":" ");
}