#1172664 (Rust) No.3594 Subset OR

• 問題: No.3594 Subset OR
• ユーザー: akakimidori
• 提出日時: 2026-07-23 21:42:23
• 言語: Rust (1.94.0 + proconio + num + itertools)
• 結果: AC
• 実行時間: 368 ms / 3,000 ms
• コード長: 3,324 bytes
• コンパイル時間: 5,672 ms
• コンパイル使用メモリ: 191,732 KB
• 実行使用メモリ: 136,576 KB
• 最終ジャッジ日時: 2026-07-23 21:42:56
• 合計ジャッジ時間: 23,783 ms
• ジャッジサーバーID （参考情報）: judge1_0 / judge2_0

ファイルパターン	結果
sample	AC * 3
other	AC * 37

コンパイルメッセージ

warning: unused import: `std::io::Write`
 --> src/main.rs:2:5
  |
2 | use std::io::Write;
  |     ^^^^^^^^^^^^^^
  |
  = note: `#[warn(unused_imports)]` (part of `#[warn(unused)]`) on by default

warning: unused variable: `b`
  --> src/main.rs:13:9
   |
13 |     let b = a.iter().map(|a| a.count_ones()).max().unwrap();
   |         ^ help: if this is intentional, prefix it with an underscore: `_b`
   |
   = note: `#[warn(unused_variables)]` (part of `#[warn(unused)]`) on by default

warning: type alias `Map` is never used
 --> src/main.rs:4:6
  |
4 | type Map<K, V> = BTreeMap<K, V>;
  |      ^^^
  |
  = note: `#[warn(dead_code)]` (part of `#[warn(unused)]`) on by default

warning: type alias `Set` is never used
 --> src/main.rs:5:6
  |
5 | type Set<T> = BTreeSet<T>;
  |      ^^^

warning: type alias `Deque` is never used
 --> src/main.rs:6:6
  |
6 | type Deque<T> = VecDeque<T>;
  |      ^^^^^

ソースコード

use std::collections::*;
use std::io::Write;

type Map<K, V> = BTreeMap<K, V>;
type Set<T> = BTreeSet<T>;
type Deque<T> = VecDeque<T>;

fn main() {
    input! {
        n: usize,
        a: [usize; n],
    }
    let b = a.iter().map(|a| a.count_ones()).max().unwrap();
    let w = 8 * std::mem::size_of::<usize>();
    let mut buf = vec![0usize; (1 << 30) / w];
    for a in a {
        buf[a / w] |= 1 << (a % w);
    }
    bitwise_transform(&mut buf, |a, b| *a |= *b);
    let mut ans = 0;
    for mut x in buf {
        x |= (x & 0xFFFFFFFF00000000) >> 32;
        x |= (x & 0xFFFF0000FFFF0000) >> 16;
        x |= (x & 0xFF00FF00FF00FF00) >> 8;
        x |= (x & 0xF0F0F0F0F0F0F0F0) >> 4;
        x |= (x & 0xCCCCCCCCCCCCCCCC) >> 2;
        x |= (x & 0xAAAAAAAAAAAAAAAA) >> 1;
        ans += x.count_ones();
    }
    println!("{}", ans);
}

// そのまま上位bitからやろうと思うと？
// 0が空の場合、重み2倍にして再帰
// 1が空ならそのまま再帰
// 両方あるとき
// 0, 1 それぞれ再帰すると両方で作れるものを重複して数える
// 1 はそのまま再帰
// 0は元の列で再帰
//
// 終わりの計算量
//
// B = max popcnt(A_i)
// を考えてみる
// 2^(30-B) 程度なら簡単
// Bが小さいときは？
// こっちは全列挙できる
//
//
// いやbitsetでゼータ変換するだけじゃん

// ---------- begin input macro ----------
// reference: https://qiita.com/tanakh/items/0ba42c7ca36cd29d0ac8
#[macro_export]
macro_rules! input {
    (source = $s:expr, $($r:tt)*) => {
        let mut iter = $s.split_whitespace();
        input_inner!{iter, $($r)*}
    };
    ($($r:tt)*) => {
        let s = {
            use std::io::Read;
            let mut s = String::new();
            std::io::stdin().read_to_string(&mut s).unwrap();
            s
        };
        let mut iter = s.split_whitespace();
        input_inner!{iter, $($r)*}
    };
}

#[macro_export]
macro_rules! input_inner {
    ($iter:expr) => {};
    ($iter:expr, ) => {};
    ($iter:expr, $var:ident : $t:tt $($r:tt)*) => {
        let $var = read_value!($iter, $t);
        input_inner!{$iter $($r)*}
    };
}

#[macro_export]
macro_rules! read_value {
    ($iter:expr, ( $($t:tt),* )) => {
        ( $(read_value!($iter, $t)),* )
    };
    ($iter:expr, [ $t:tt ; $len:expr ]) => {
        (0..$len).map(|_| read_value!($iter, $t)).collect::<Vec<_>>()
    };
    ($iter:expr, chars) => {
        read_value!($iter, String).chars().collect::<Vec<char>>()
    };
    ($iter:expr, bytes) => {
        read_value!($iter, String).bytes().collect::<Vec<u8>>()
    };
    ($iter:expr, usize1) => {
        read_value!($iter, usize) - 1
    };
    ($iter:expr, $t:ty) => {
        $iter.next().unwrap().parse::<$t>().expect("Parse error")
    };
}
// ---------- end input macro ----------
// ---------- begin bitwise transform ----------
pub fn bitwise_transform<T, F>(a: &mut [T], mut f: F)
where
    F: FnMut(&mut T, &mut T)
{
    let n = a.len().trailing_zeros() as usize;
    assert!(a.len() == 1 << n);
    for i in 0..n {
        for a in a.chunks_exact_mut(2 << i) {
            let (x, y) = a.split_at_mut(1 << i);
            for (x, y) in x.iter_mut().zip(y) {
                f(x, y);
            }
        }
    }
}
// ---------- end bitwise transform ----------