結果

問題 No.206 数の積集合を求めるクエリ
ユーザー te-shte-sh
提出日時 2017-05-12 11:24:07
言語 D
(dmd 2.106.1)
結果
AC  
実行時間 149 ms / 7,000 ms
コード長 1,277 bytes
コンパイル時間 1,954 ms
コンパイル使用メモリ 167,680 KB
実行使用メモリ 29,068 KB
最終ジャッジ日時 2024-06-12 19:12:40
合計ジャッジ時間 5,001 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
6,816 KB
testcase_01 AC 1 ms
6,940 KB
testcase_02 AC 2 ms
6,944 KB
testcase_03 AC 1 ms
6,944 KB
testcase_04 AC 1 ms
6,940 KB
testcase_05 AC 1 ms
6,940 KB
testcase_06 AC 5 ms
6,940 KB
testcase_07 AC 4 ms
6,944 KB
testcase_08 AC 5 ms
6,940 KB
testcase_09 AC 5 ms
6,944 KB
testcase_10 AC 1 ms
6,940 KB
testcase_11 AC 1 ms
6,940 KB
testcase_12 AC 6 ms
6,940 KB
testcase_13 AC 5 ms
6,940 KB
testcase_14 AC 5 ms
6,940 KB
testcase_15 AC 4 ms
6,940 KB
testcase_16 AC 5 ms
6,944 KB
testcase_17 AC 123 ms
28,416 KB
testcase_18 AC 119 ms
27,176 KB
testcase_19 AC 119 ms
28,016 KB
testcase_20 AC 108 ms
22,732 KB
testcase_21 AC 113 ms
27,340 KB
testcase_22 AC 113 ms
25,948 KB
testcase_23 AC 119 ms
26,568 KB
testcase_24 AC 149 ms
27,852 KB
testcase_25 AC 148 ms
28,652 KB
testcase_26 AC 136 ms
26,108 KB
testcase_27 AC 124 ms
25,960 KB
testcase_28 AC 139 ms
29,068 KB
testcase_29 AC 137 ms
27,424 KB
testcase_30 AC 134 ms
26,544 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std.algorithm, std.conv, std.range, std.stdio, std.string;
import std.bitmanip;  // BitArray
import std.math;      // math functions
import std.numeric;   // gcd, fft

void main()
{
  auto rd = readln.split.to!(size_t[]), l = rd[0], m = rd[1], n = rd[2];
  auto ai = readln.split.to!(int[]);
  auto bi = readln.split.to!(int[]);
  auto q = readln.chomp.to!int;

  calc2(n, ai, bi, q);
}

auto calc1(size_t n, int[] ai, int[] bi, int q)
{
  auto bai = BitArray(); bai.length = n+1;
  foreach (a; ai) bai[a] = true;
  auto bbi = BitArray(); bbi.length = n+1;
  foreach (b; bi) bbi[b] = true;

  foreach (_; 0..q) {
    auto bci = bai & bbi;
    writeln((cast(size_t[])(bci)).map!(i => i.popcnt).sum);
    bbi <<= 1;
  }
}

auto calc2(size_t n, int[] ai, int[] bi, int q)
{
  auto nf = (1 << ((n * 2).bsr + 1));
  auto cai = new double[](nf); cai[] = 0;
  foreach (a; ai) cai[a] = 1;
  auto cbi = new double[](nf); cbi[] = 0;
  foreach (b; bi) cbi[n-b] = 1;

  auto fai = fft(cai), fbi = fft(cbi);
  fai[] *= fbi[];

  auto ri = inverseFft(fai);
  foreach (r; ri[n..n+q])
    writeln(r.re.round.to!int);
}

pragma(inline) {
  import core.bitop;
  pure int bsr(T)(T n) { return core.bitop.bsr(ulong(n)); }
  pure int popcnt(T)(T n) { return core.bitop.popcnt(ulong(n)); }
}
0