結果

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

ソースコード

diff # 

#include<cstdio>
using namespace std;
#include<complex>
#include<vector>
void dft(vector<complex<double> >&A,int sign)
{
	int N=A.size()>>1;
	if(N==0)return;
	vector<complex<double> >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);
	complex<double>z(cos(M_PI/N),sin(M_PI/N)*sign),p=1;
	for(int i=0;i<N;i++)
	{
		A[i]=F[i]+p*G[i];
		A[i+N]=F[i]-p*G[i];
		p*=z;
	}
}
vector<long long>multiply(const vector<int>&A,const vector<int>&B)
{
	if(A.empty()||B.empty())
	{
		return(vector<long long>){};
	}
	int N=1;
	vector<long long>ret(A.size()+B.size()-1);
	while(N<ret.size())N<<=1;
	vector<complex<double> >F(N),G(N);
	for(int i=0;i<A.size();i++)F[i]=A[i];
	for(int i=0;i<B.size();i++)G[i]=B[i];
	dft(F,1);
	dft(G,1);
	for(int i=0;i<N;i++)F[i]*=G[i];
	dft(F,-1);
	for(int i=0;i<ret.size();i++)ret[i]=llround((F[i]/=N).real());
	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;
	}
	vector<long long>C=multiply(A,B);
	int Q;scanf("%d",&Q);
	for(int i=0;i<Q;i++)
	{
		printf("%d\n",(int)C[N+i]);
	}
}
0