結果

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

ソースコード

diff # 

#pragma GCC optimize ("O3")
#pragma GCC target ("avx")
#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) {
        init_montgomery_reduction();
    }

    uint32_t iv;
    int r2;

    int reduce(uint64_t x) {
        uint64_t y = uint64_t(uint32_t(x) * iv) * mod;
        int res = int(x >> 32) - int(y >> 32);
        return res < 0 ? res + mod : res;
    }

    int transform(int n) {
        return reduce(int64_t(n) * r2);
    }

    int normalize(int n) {
        return n >= mod ? n - mod : n;
    }

    void init_montgomery_reduction() {
        iv = 1;
        for (int i = 0; i < 5; ++i) iv *= 2 - iv * uint32_t(mod);
        r2 = -uint64_t(mod) % mod;
    }

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

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

    int pow(int x, int y) {
        int res = 1;
        while (y > 0) {
            if (y & 1) res = int64_t(res) * x % mod;
            x = int64_t(x) * x % mod;
            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);
            w = transform(w);
            for (int j = 0; j < n; j += i * 2) {
                int wn = transform(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 = transform(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);
        for (int i = 0; i < t; i++) {
            a[i] = transform(a[i]);
            b[i] = transform(b[i]);
        }
        ntt(a);
        ntt(b);
        for (int i = 0; i < t; i++) {
            a[i] = mul(a[i], b[i]);
        }
        ntt(a, true);
        a.resize(s);
        for (int i = 0; i < s; i++) {
            a[i] = reduce(a[i]);
        }
        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