結果

問題 No.206 数の積集合を求めるクエリ
ユーザー はまやんはまやんはまやんはまやん
提出日時 2017-05-20 11:56:21
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 132 ms / 7,000 ms
コード長 2,189 bytes
コンパイル時間 1,993 ms
コンパイル使用メモリ 173,840 KB
実行使用メモリ 14,000 KB
最終ジャッジ日時 2024-09-19 00:24:58
合計ジャッジ時間 4,863 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 5 ms
5,376 KB
testcase_07 AC 4 ms
5,376 KB
testcase_08 AC 4 ms
5,376 KB
testcase_09 AC 4 ms
5,376 KB
testcase_10 AC 2 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 5 ms
5,376 KB
testcase_13 AC 5 ms
5,376 KB
testcase_14 AC 5 ms
5,376 KB
testcase_15 AC 5 ms
5,376 KB
testcase_16 AC 5 ms
5,376 KB
testcase_17 AC 123 ms
13,748 KB
testcase_18 AC 115 ms
14,000 KB
testcase_19 AC 123 ms
13,868 KB
testcase_20 AC 115 ms
13,872 KB
testcase_21 AC 113 ms
13,996 KB
testcase_22 AC 118 ms
13,872 KB
testcase_23 AC 121 ms
13,872 KB
testcase_24 AC 132 ms
13,872 KB
testcase_25 AC 128 ms
13,876 KB
testcase_26 AC 123 ms
13,876 KB
testcase_27 AC 117 ms
13,652 KB
testcase_28 AC 124 ms
13,804 KB
testcase_29 AC 125 ms
13,872 KB
testcase_30 AC 119 ms
13,872 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=a;i<b;i++)
using namespace std; void _main(); int main() { cin.tie(0); ios::sync_with_stdio(false); _main(); }
//---------------------------------------------------------------------------------------------------
const double PI = acos(-1.0);
typedef complex<float> P;
void fft(vector<P> &a, int isRev = 0) { // 配列aのサイズは2の累乗じゃないとダメ
    const P I(0.0, 1.0); int n = a.size(); float theta = 2 * PI / n; if (isRev) theta = -2 * PI / n;
    for (int m = n; m >= 2; m >>= 1) { int mh = m >> 1; rep(i, 0, mh) { P w = exp(i*theta*I);
    for (int j = i; j < n; j += m) { int k = j + mh; P x = a[j] - a[k]; a[j] += a[k]; a[k] = w * x; } }
        theta *= 2; } int i = 0;
    rep(j, 1, n - 1) { for (int k = n >> 1; k >(i ^= k); k >>= 1); if (j < i) swap(a[i], a[j]); }
    if (isRev) rep(i, 0, n) a[i] /= n; }
vector<int> convolution(vector<int> a, vector<int> b) {
    int n = a.size(), m = 1; while (m < n) m *= 2;
    vector<P> A(m), B(m); rep(i, 0, n) { A[i] = P(a[i], 0); B[i] = P(b[i], 0); } fft(A); fft(B);
    vector<P> C(m); rep(i, 0, m) C[i] = A[i] * B[i]; fft(C, 1);
    vector<int> res; rep(i, 0, n) res.push_back(int(real(C[i]) + 0.3)); return res; }
/*---------------------------------------------------------------------------------------------------
　　　　　　　　　　　 ∧＿∧  
　　　　　 ∧＿∧ 　（´<_｀ ）　 Welcome to My Coding Space!
　　　　 （ ´_ゝ`）　/　 ⌒i     
　　　　／　　　＼　 　  |　|     
　　　 /　　 /￣￣￣￣/　　|  
　 ＿_(__ﾆつ/　    ＿/ .| .|＿＿＿＿  
　 　　　＼/＿＿＿＿/　（u　⊃  
---------------------------------------------------------------------------------------------------*/




int L, M, N, Q;
//---------------------------------------------------------------------------------------------------
void _main() {
    cin >> L >> M >> N;
    vector<int> A(N * 2 + 2), B(N * 2 + 2);
    int x;
    rep(i, 0, L) cin >> x, A[x] = 1;
    rep(i, 0, M) cin >> x, B[N - x] = 1;
    cin >> Q;

    auto ans = convolution(A, B);
    rep(i, N, N + Q) printf("%d\n", ans[i]);
}
0