結果

問題 No.206 数の積集合を求めるクエリ
ユーザー pekempeypekempey
提出日時 2016-10-30 19:31:19
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 91 ms / 7,000 ms
コード長 3,406 bytes
コンパイル時間 2,184 ms
コンパイル使用メモリ 178,800 KB
実行使用メモリ 6,876 KB
最終ジャッジ日時 2024-05-03 19:56:20
合計ジャッジ時間 5,889 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 75 ms
6,816 KB
testcase_01 AC 73 ms
6,748 KB
testcase_02 AC 74 ms
6,872 KB
testcase_03 AC 74 ms
6,872 KB
testcase_04 AC 74 ms
6,660 KB
testcase_05 AC 79 ms
6,876 KB
testcase_06 AC 74 ms
6,744 KB
testcase_07 AC 74 ms
6,744 KB
testcase_08 AC 71 ms
6,872 KB
testcase_09 AC 72 ms
6,744 KB
testcase_10 AC 72 ms
6,748 KB
testcase_11 AC 73 ms
6,868 KB
testcase_12 AC 75 ms
6,872 KB
testcase_13 AC 76 ms
6,744 KB
testcase_14 AC 71 ms
6,772 KB
testcase_15 AC 75 ms
6,872 KB
testcase_16 AC 76 ms
6,872 KB
testcase_17 AC 77 ms
6,868 KB
testcase_18 AC 76 ms
6,744 KB
testcase_19 AC 77 ms
6,872 KB
testcase_20 AC 74 ms
6,740 KB
testcase_21 AC 73 ms
6,876 KB
testcase_22 AC 75 ms
6,744 KB
testcase_23 AC 78 ms
6,868 KB
testcase_24 AC 91 ms
6,876 KB
testcase_25 AC 85 ms
6,744 KB
testcase_26 AC 82 ms
6,740 KB
testcase_27 AC 81 ms
6,740 KB
testcase_28 AC 86 ms
6,872 KB
testcase_29 AC 86 ms
6,740 KB
testcase_30 AC 82 ms
6,872 KB
権限があれば一括ダウンロードができます

ソースコード

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