結果

問題 No.206 数の積集合を求めるクエリ
ユーザー yuppe19 😺yuppe19 😺
提出日時 2018-08-17 09:04:04
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 240 ms / 7,000 ms
コード長 1,982 bytes
コンパイル時間 1,064 ms
コンパイル使用メモリ 87,104 KB
実行使用メモリ 15,216 KB
最終ジャッジ日時 2024-04-16 18:12:16
合計ジャッジ時間 5,615 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 3 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 7 ms
5,376 KB
testcase_07 AC 8 ms
5,376 KB
testcase_08 AC 8 ms
5,376 KB
testcase_09 AC 8 ms
5,376 KB
testcase_10 AC 2 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 8 ms
5,376 KB
testcase_13 AC 7 ms
5,376 KB
testcase_14 AC 7 ms
5,376 KB
testcase_15 AC 7 ms
5,376 KB
testcase_16 AC 7 ms
5,376 KB
testcase_17 AC 230 ms
15,212 KB
testcase_18 AC 218 ms
15,216 KB
testcase_19 AC 224 ms
15,036 KB
testcase_20 AC 218 ms
15,088 KB
testcase_21 AC 222 ms
15,084 KB
testcase_22 AC 221 ms
15,088 KB
testcase_23 AC 227 ms
15,084 KB
testcase_24 AC 237 ms
15,212 KB
testcase_25 AC 240 ms
15,168 KB
testcase_26 AC 229 ms
15,212 KB
testcase_27 AC 219 ms
15,104 KB
testcase_28 AC 230 ms
15,084 KB
testcase_29 AC 231 ms
15,084 KB
testcase_30 AC 226 ms
15,088 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <map>
#include <vector>
using namespace std;
using i64 = long long;

inline i64 mod_mul(const i64& x, const i64& y, const i64& m) {
  return x * y % m;
}

inline i64 mod_pow(const i64& a, const i64& r, const i64& n, const i64& m) {
  if(n == 0) { return r; }
  if(n & 1)  { return mod_pow(a, mod_mul(a, r, m), n-1, m); }
  return mod_pow(mod_mul(a, a, m), r, n>>1, m);
}

map<tuple<i64, i64, i64>, i64> cache;

inline i64 mod_pow(const i64 a, const i64 n, const i64 m) {
  if(cache.count({a, n, m})) { return cache[{a, n, m}]; }
  return cache[{a, n, m}] = mod_pow(a, 1, n, m);
}

constexpr i64 O = 3;
constexpr i64 M = 998244353;

void myfmt(vector<i64> &a, bool inv) {
  int n = int(a.size());
  if(n == 1) { return; }
  int m = n / 2;
  vector<i64> a0(m), a1(m);
  for(int i=0, j=0; i<m; ++i) {
    a0[i] = a[j++];
    a1[i] = a[j++];
  }
  myfmt(a0, inv);
  myfmt(a1, inv);
  i64 z = mod_pow(O, (M-1)/n, M);
  if(inv) { z = mod_pow(z, M-2, M); }
  i64 pz = 1;
  for(int i=0; i<n; ++i) {
    a[i] = a0[i%m] + mod_mul(pz, a1[i%m], M);
    a[i] %= M;
    pz = mod_mul(pz, z, M);
  }
}

void fmt(vector<i64> &a) {
  myfmt(a, false);
}

void ifmt(vector<i64> &a) {
  myfmt(a, true);
  int n = int(a.size());
  i64 inv = mod_pow(n, M-2, M);
  for(int i=0; i<n; ++i) {
    a[i] = mod_mul(a[i], inv, M);
  }
}

vector<i64> convol(vector<i64> a, vector<i64> b) {
  int n = 1;
  while(n < a.size() + b.size()) { n <<= 1; }
  a.resize(n);
  b.resize(n);
  fmt(a);
  fmt(b);
  vector<i64> c(n);
  for(int i=0; i<n; ++i) {
    c[i] = mod_mul(a[i], b[i], M);
  }
  ifmt(c);
  return c;
}

int main(void) {
  int L, M, N; scanf("%d%d%d", &L, &M, &N);
  int n = N + 10;
  vector<i64> p(n), q(n);
  for(int i=0; i<L; ++i) { int x; scanf("%d", &x); p[x]   = true; }
  for(int i=0; i<M; ++i) { int x; scanf("%d", &x); q[N-x] = true; }
  vector<i64> r = convol(p, q);
  int Q; scanf("%d", &Q);
  for(int v=0; v<Q; ++v) {
    printf("%lld\n", r[N+v]);
  }
  return 0;
}
0