結果

問題 No.206 数の積集合を求めるクエリ
コンテスト
ユーザー pekempey
提出日時 2016-10-30 19:14:06
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 195 ms / 7,000 ms
コード長 2,512 bytes
コンパイル時間 1,396 ms
コンパイル使用メモリ 163,772 KB
実行使用メモリ 6,872 KB
最終ジャッジ日時 2024-11-24 23:50:30
合計ジャッジ時間 9,007 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define getchar getchar_unlocked

struct NumberTheoreticTransform {
    int mod;
    int root;

    NumberTheoreticTransform(int mod, int root) : mod(mod), root(root) {}

    int mul(int x, int y) {
        return int64_t(x) * y % mod;
    }

    int add(int x, int y) {
        return (x += y) >= mod ? x - mod : x;
    }

    int pow(int x, int y) {
        int res = 1;
        while (y > 0) {
            if (y & 1) res = mul(res, x);
            x = mul(x, x);
            y >>= 1;
        }
        return res;
    }

    int inv(int x) {
        return pow(x, mod - 2);
    }

    void ntt(std::vector<int> &a, bool rev = false) {
        int n = a.size();
        int h = 0;
        for (int i = 0; 1 << i < n; i++) h++;
        for (int i = 0; i < n; i++) {
            int j = 0;
            for (int k = 0; k < h; k++) j |= (i >> k & 1) << (h - 1 - k);
            if (i < j) std::swap(a[i], a[j]);
        }
        for (int i = 1; i < n; i *= 2) {
            int w = pow(root, (mod - 1) / (i * 2));
            if (rev) w = inv(w);
            for (int j = 0; j < n; j += i * 2) {
                int wn = 1;
                for (int k = 0; k < i; k++) {
                    int s = a[j + k + 0];
                    int t = mul(a[j + k + i], wn);
                    a[j + k + 0] = add(s, t);
                    a[j + k + i] = add(s, mod - t);
                    wn = mul(wn, w);
                }
            }
        }
        int v = inv(n);
        if (rev) for (int i = 0; i < n; i++) a[i] = mul(a[i], v);
    }

    std::vector<int> mul(std::vector<int> a, std::vector<int> b) {
        int s = a.size() + b.size() - 1;
        int t = 1;
        while (t < s) t *= 2;
        a.resize(t);
        b.resize(t);
        ntt(a);
        ntt(b);
        for (int i = 0; i < t; i++) {
            a[i] = mul(a[i], b[i]);
        }
        ntt(a, true);
        a.resize(s);
        return a;
    }
};

int in() {
    char c;
    while ((c = getchar()) < '0') if (c == EOF) return -1;
    int n = c - '0';
    while ((c = getchar()) >= '0') n = n * 10 + (c - '0');
    return n;
}

int main() {
    int L = in(), M = in(), N = in();
    const int X = 1 << 17;
    vector<int> a(X), b(X);
    for (int i = 0; i < L; i++) a[in()] = 1;
    for (int i = 0; i < M; i++) b[N - in()] = 1;
    NumberTheoreticTransform ntt(1012924417, 5);
    a = ntt.mul(a, b);
    int Q = in();
    for (int i = 0; i < Q; i++) {
        printf("%d\n", a[N + i]);
    }
}
0