結果

問題 No.206 数の積集合を求めるクエリ
ユーザー leaf_1415leaf_1415
提出日時 2019-01-11 02:48:41
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 473 ms / 7,000 ms
コード長 1,824 bytes
コンパイル時間 854 ms
コンパイル使用メモリ 71,324 KB
実行使用メモリ 47,400 KB
最終ジャッジ日時 2024-11-25 03:49:25
合計ジャッジ時間 13,939 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 304 ms
44,408 KB
testcase_01 AC 308 ms
44,400 KB
testcase_02 AC 306 ms
44,408 KB
testcase_03 AC 299 ms
46,308 KB
testcase_04 AC 305 ms
46,444 KB
testcase_05 AC 305 ms
46,444 KB
testcase_06 AC 308 ms
44,444 KB
testcase_07 AC 303 ms
44,480 KB
testcase_08 AC 303 ms
44,360 KB
testcase_09 AC 303 ms
44,484 KB
testcase_10 AC 302 ms
44,276 KB
testcase_11 AC 302 ms
44,408 KB
testcase_12 AC 311 ms
44,496 KB
testcase_13 AC 312 ms
44,468 KB
testcase_14 AC 305 ms
46,472 KB
testcase_15 AC 298 ms
46,340 KB
testcase_16 AC 299 ms
44,456 KB
testcase_17 AC 350 ms
47,384 KB
testcase_18 AC 328 ms
46,608 KB
testcase_19 AC 353 ms
47,360 KB
testcase_20 AC 334 ms
45,960 KB
testcase_21 AC 336 ms
46,424 KB
testcase_22 AC 334 ms
46,360 KB
testcase_23 AC 353 ms
47,400 KB
testcase_24 AC 473 ms
47,384 KB
testcase_25 AC 472 ms
47,268 KB
testcase_26 AC 455 ms
46,740 KB
testcase_27 AC 403 ms
45,804 KB
testcase_28 AC 443 ms
46,928 KB
testcase_29 AC 441 ms
46,600 KB
testcase_30 AC 429 ms
45,724 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <complex>
#include <vector>
#define llint long long 
#define PI 3.141592653589793238

using namespace std;
typedef complex<double> Complex;

void DFT(vector<Complex> &f, int n, vector<Complex> &F)
{
	F.resize(n);
	if(n == 1){
		F[0] = f[0];
		return;
	}
	
	vector<Complex> f0(n/2), f1(n/2), F0, F1;
	
	for(int i = 0; i < n/2; i++) f0[i] = f[i*2];
	DFT(f0, n/2, F0);
	for(int i = 0; i < n/2; i++) f1[i] = f[i*2+1];
	DFT(f1, n/2, F1);
	
	Complex z = Complex(cos(2*PI/n), sin(2*PI/n)), x = Complex(1, 0);
	for(int i = 0; i < n; i++){
		F[i] = F0[i%(n/2)] + x * F1[i%(n/2)];
		x *= z;
	}
}

//The answer must be divided by n.
void IDFT(vector<Complex> &f, int n, vector<Complex> &F)
{
	F.resize(n);
	if(n == 1){
		F[0] = f[0];
		return;
	}
	
	vector<Complex> f0(n/2), f1(n/2), F0, F1;
	
	for(int i = 0; i < n/2; i++) f0[i] = f[i*2];
	IDFT(f0, n/2, F0);
	for(int i = 0; i < n/2; i++) f1[i] = f[i*2+1];
	IDFT(f1, n/2, F1);
	
	Complex z = Complex(cos(2*PI/n), -sin(2*PI/n)), x = Complex(1, 0);
	for(int i = 0; i < n; i++){
		F[i] = F0[i%(n/2)] + x * F1[i%(n/2)];
		x *= z;
	}
}

llint round(Complex c){
	return (int)(c.real()+0.5);
}

llint L, M, N, Q;
llint A[100005], B[100005];
llint a[100005], b[100005];
const llint S = 1<<18;
vector<Complex> f(S), g(S), h, F, G, H(S);

int main(void)
{
	cin >> L >> M >> N;
	for(int i = 0; i < L; i++) cin >> A[i];
	for(int i = 0; i < M; i++) cin >> B[i];
	cin >> Q;
	
	for(int i = 0; i < L; i++) a[A[i]] = 1;
	for(int i = 0; i < M; i++) b[B[i]] = 1;
	
	for(int i = 1; i <= 100000; i++) f[100001-i] = a[i];
	for(int i = 1; i <= 100000; i++) g[i] = b[i];
	DFT(f, S, F), DFT(g, S, G);
	for(int i = 0; i < S; i++) H[i] = F[i]*G[i];
	IDFT(H, S, h);
	for(int i = 0; i < S; i++) h[i] /= S;
	for(int i = 0; i < Q; i++) cout << round(h[100001-i]) << endl;
	
	return 0;
}
0