結果

問題 No.3594 Subset OR
ユーザー Ryuhei Mori
提出日時 2026-07-25 15:10:39
言語 C++23(gcc16)
(gcc 16.1.0 + boost 1.90.0)
コンパイル:
g++-16 -O2 -lm -std=c++23 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
AC  
実行時間 447 ms / 3,000 ms
+ 669µs
コード長 1,214 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,206 ms
コンパイル使用メモリ 82,856 KB
実行使用メモリ 69,924 KB
最終ジャッジ日時 2026-07-25 15:11:04
合計ジャッジ時間 22,517 ms
ジャッジサーバーID
（参考情報） 		judge1_0 / judge2_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 37
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#pragma GCC optimize("O3,unroll-loops")
#pragma GCC target("avx2,popcnt")

#include <cstdio>
#include <array>
#include <vector>

using u64 = unsigned long long;
using u32 = unsigned;

using bs_t = std::array<u64, 1<<(30-7)>;

bs_t B;

void set(bs_t &B, u32 i){
  B[i/64] |= 1ULL << (i%64);
}

void zeta(bs_t &B){
  for(u32 i = 1; i < B.size(); i <<= 1){
    for(u32 j = 0; j < B.size(); j++){
      if(j & i){
        B[j - i] |= B[j];
      }
    }
  }
  for(u32 j = 0; j < B.size(); j++){
    B[j] |= B[j] >> 32;
    B[j] |= (B[j] >> 16) & 0x0000FFFF0000FFFF;
    B[j] |= (B[j] >> 8) & 0x00FF00FF00FF00FF;
    B[j] |= (B[j] >> 4) & 0x0F0F0F0F0F0F0F0F;
    B[j] |= (B[j] >> 2) & 0x3333333333333333;
    B[j] |= (B[j] >> 1) & 0x5555555555555555;
  }
}

u32 sum(bs_t &B){
  u32 ans = 0;
  for(u64 b: B){
    ans += __builtin_popcountll(b);
  }
  return ans;
}

int main(){
  u32 n;
  scanf("%u", &n);
  std::vector<u32> A;
  for(u32 i = 0; i < n; i++){
    u32 a;
    scanf("%u", &a);
    if(a >= (1<<29)){
      set(B, a - (1<<29));
    }
    else {
      A.push_back(a);
    }
  }
  zeta(B);
  u32 ans = sum(B);
  for(u32 a: A){
    set(B, a);
  }
  zeta(B);
  ans += sum(B);
  printf("%u\n", ans);
  return 0;
}
0