結果

問題 No.206 数の積集合を求めるクエリ
ユーザー kotatsugamekotatsugame
提出日時 2019-09-26 18:13:12
言語 C++14
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 178 ms / 7,000 ms
コード長 1,209 bytes
コンパイル時間 812 ms
コンパイル使用メモリ 84,580 KB
実行使用メモリ 21,828 KB
最終ジャッジ日時 2023-10-24 15:38:02
合計ジャッジ時間 5,195 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,348 KB
testcase_01 AC 2 ms
4,348 KB
testcase_02 AC 2 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 5 ms
4,348 KB
testcase_07 AC 5 ms
4,348 KB
testcase_08 AC 6 ms
4,348 KB
testcase_09 AC 6 ms
4,348 KB
testcase_10 AC 2 ms
4,348 KB
testcase_11 AC 2 ms
4,348 KB
testcase_12 AC 6 ms
4,348 KB
testcase_13 AC 6 ms
4,348 KB
testcase_14 AC 6 ms
4,348 KB
testcase_15 AC 6 ms
4,348 KB
testcase_16 AC 6 ms
4,348 KB
testcase_17 AC 169 ms
21,828 KB
testcase_18 AC 160 ms
21,828 KB
testcase_19 AC 169 ms
21,828 KB
testcase_20 AC 159 ms
21,828 KB
testcase_21 AC 163 ms
21,828 KB
testcase_22 AC 161 ms
21,828 KB
testcase_23 AC 168 ms
21,828 KB
testcase_24 AC 178 ms
21,828 KB
testcase_25 AC 176 ms
21,828 KB
testcase_26 AC 168 ms
21,828 KB
testcase_27 AC 162 ms
21,828 KB
testcase_28 AC 170 ms
21,828 KB
testcase_29 AC 170 ms
21,828 KB
testcase_30 AC 166 ms
21,828 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
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