結果

問題 No.206 数の積集合を求めるクエリ
コンテスト
ユーザー akakimidori
提出日時 2019-04-27 02:19:47
言語 C
(gcc 13.3.0)
結果
AC  
実行時間 423 ms / 7,000 ms
コード長 1,248 bytes
コンパイル時間 174 ms
コンパイル使用メモリ 30,976 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-26 02:01:48
合計ジャッジ時間 4,329 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<stdio.h>
#include<stdlib.h>
#include<stdint.h>
#include<inttypes.h>

typedef int32_t i32;
typedef uint64_t u64;

static inline i32 popcnt (u64 n, const uint8_t *cnt) {
  return cnt[n & 0xFFFF] + cnt[(n >> 16) & 0xFFFF] + cnt[(n >> 32) & 0xFFFF] + cnt[(n >> 48) & 0xFFFF];
}

void run (void) {
  i32 l, m, n;
  scanf ("%" SCNi32 "%" SCNi32 "%" SCNi32, &l, &m, &n);
  const i32 f = 8 * sizeof (u64);
  i32 len = (n + 1) / f + 1;
  u64 *a = (u64 *) calloc (len, sizeof (u64));
  u64 *b = (u64 *) calloc (len, sizeof (u64));
  while (l--) {
    i32 v;
    scanf ("%" SCNi32, &v);
    a[v / f] |= (u64)1 << (v % f);
  }
  while (m--) {
    i32 v;
    scanf ("%" SCNi32, &v);
    b[v / f] |= (u64)1 << (v % f);
  }
  uint8_t *cnt = (uint8_t *) calloc (1 << 16, sizeof (uint8_t));
  for (i32 i = 0; i < (1 << 16); ++i) {
    i32 t = i;
    while (t > 0) {
      cnt[i]++;
      t -= t & -t;
    }
  }
  i32 q;
  scanf ("%" SCNi32, &q);
  while (q--) {
    i32 ans = 0;
    for (i32 j = 0; j < len; ++j) {
      ans += popcnt (a[j] & b[j], cnt);
    }
    printf ("%" PRIi32 "\n", ans);
    for (i32 j = len - 1; j > 0; --j) {
      b[j] = (b[j] << 1) | (b[j - 1] >> (f - 1));
    }
    b[0] <<= 1;
  }
}

int main (void) {
  run();
  return 0;
}
0