結果

問題 No.206 数の積集合を求めるクエリ
ユーザー kotatsugame
提出日時 2019-09-26 18:14:18
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 241 ms / 7,000 ms
コード長 1,432 bytes
コンパイル時間 458 ms
コンパイル使用メモリ 52,736 KB
実行使用メモリ 8,688 KB
最終ジャッジ日時 2024-09-23 08:03:41
合計ジャッジ時間 5,051 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:59:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   59 | main()
      | ^~~~

ソースコード

diff # 

#include<cstdio>
using namespace std;
//998244353,3
#include<vector>
template<int mod,int proot>
struct NTT{
	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 N=A.size()>>1;
		if(N==0)return;
		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);
		dft(G,sign);
		long long z=power(proot,sign?(long long)(mod-1)/N/2*(mod-2)%(mod-1):(mod-1)/N/2),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;
		vector<int>ret(A.size()+B.size()-1);
		while(N<ret.size())N<<=1;
		A.resize(N);
		B.resize(N);
		dft(A,false);
		dft(B,false);
		for(int i=0;i<N;i++)A[i]=(long long)A[i]*B[i]%mod;
		dft(A,true);
		long long invN=power(N,mod-2);
		for(int i=0;i<ret.size();i++)ret[i]=invN*A[i]%mod;
		return ret;
	}
};
main()
{
	int L,M,N;
	scanf("%d%d%d",&L,&M,&N);
	vector<int>A(N+1),B(N+1);
	for(int i=0;i<L;i++)
	{
		int a;scanf("%d",&a);
		A[a]=1;
	}
	for(int i=0;i<M;i++)
	{
		int a;scanf("%d",&a);
		B[N-a]=1;
	}
	NTT<998244353,3>ntt;
	vector<int>C=ntt.multiply(A,B);
	int Q;scanf("%d",&Q);
	for(int i=0;i<Q;i++)
	{
		printf("%d\n",C[N+i]);
	}
}
0