結果

問題 No.1068 #いろいろな色 / Red and Blue and more various colors (Hard)
ユーザー kotatsugamekotatsugame
提出日時 2020-05-29 21:48:09
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 2,167 ms / 3,500 ms
コード長 1,809 bytes
コンパイル時間 980 ms
コンパイル使用メモリ 76,956 KB
実行使用メモリ 33,260 KB
最終ジャッジ日時 2024-04-23 21:19:01
合計ジャッジ時間 42,070 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 4 ms
9,728 KB
testcase_01 AC 6 ms
9,600 KB
testcase_02 AC 5 ms
9,728 KB
testcase_03 AC 40 ms
10,240 KB
testcase_04 AC 28 ms
9,984 KB
testcase_05 AC 33 ms
10,112 KB
testcase_06 AC 25 ms
9,856 KB
testcase_07 AC 24 ms
9,856 KB
testcase_08 AC 31 ms
10,240 KB
testcase_09 AC 34 ms
10,112 KB
testcase_10 AC 16 ms
9,856 KB
testcase_11 AC 23 ms
9,984 KB
testcase_12 AC 14 ms
9,984 KB
testcase_13 AC 2,158 ms
33,256 KB
testcase_14 AC 2,157 ms
33,128 KB
testcase_15 AC 2,154 ms
33,260 KB
testcase_16 AC 2,155 ms
33,128 KB
testcase_17 AC 2,155 ms
33,124 KB
testcase_18 AC 2,160 ms
33,128 KB
testcase_19 AC 2,160 ms
33,256 KB
testcase_20 AC 2,157 ms
33,256 KB
testcase_21 AC 2,160 ms
33,132 KB
testcase_22 AC 2,160 ms
33,256 KB
testcase_23 AC 2,167 ms
33,256 KB
testcase_24 AC 2,152 ms
33,124 KB
testcase_25 AC 2,157 ms
33,228 KB
testcase_26 AC 2,166 ms
33,256 KB
testcase_27 AC 2,164 ms
33,128 KB
testcase_28 AC 2,164 ms
33,128 KB
testcase_29 AC 2,097 ms
33,128 KB
testcase_30 AC 2,099 ms
33,256 KB
testcase_31 AC 4 ms
9,600 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:79:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   79 | main()
      | ^~~~

ソースコード

diff # 

#include<iostream>
using namespace std;
//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
	NTT()
	{
		int sz=23;
		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;
		}
	}
	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;
		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;
	}
};
const long mod=998244353;
int N,Q;
long A[2<<17];
vector<int>dp[2<<17];
main()
{
	cin>>N>>Q;
	for(int i=0;i<N;i++)
	{
		cin>>A[i];
		A[i]--;
		A[i]%=mod;
		dp[i]=(vector<int>){(int)A[i],1};
	}
	NTT<998244353,3>P;
	for(int k=0;1<<k<N;k++)
	{
		for(int i=0;i+(1<<k)<N;i+=2<<k)
		{
			dp[i]=P.multiply(dp[i],dp[i+(1<<k)]);
		}
	}
	for(int i=0;i<Q;i++)
	{
		int b;cin>>b;
		cout<<dp[0][b]<<endl;
	}
}
0