結果

問題 No.1240 Or Sum of Xor Pair
コンテスト
ユーザー akakimidori
提出日時 2020-10-31 14:28:44
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 79 ms / 2,000 ms
コード長 1,618 bytes
コンパイル時間 16,999 ms
コンパイル使用メモリ 387,236 KB
実行使用メモリ 5,504 KB
最終ジャッジ日時 2024-07-22 04:59:09
合計ジャッジ時間 20,137 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

use std::io::Read;

fn run() {
    let mut s = String::new();
    std::io::stdin().read_to_string(&mut s).unwrap();
    let mut it = s.trim().split_whitespace();
    it.next();
    let sup: u64 = it.next().unwrap().parse().unwrap();
    let mut a: Vec<u64> = it.map(|s| s.parse().unwrap()).collect();
    a.sort();
    let a = a.as_slice();
    let calc = |a: &[u64], b: &[u64]| -> u64 {
        let mut cnt = [0; 18];
        for a in a.iter() {
            for (i, c) in cnt.iter_mut().enumerate() {
                *c += *a >> i & 1;
            }
        }
        let n = a.len() as u64;
        let mut sum = 0u64;
        for b in b.iter() {
            for (i, c) in cnt.iter().enumerate() {
                sum += (*c + (n - *c) * (*b >> i & 1)) << i;
            }
        }
        sum
    };
    let mut dfs = vec![(a, a, 20)];
    let mut ans = 0u64;
    while let Some((a, b, k)) = dfs.pop() {
        if a.is_empty() || b.is_empty() || k == 0 {
            continue;
        }
        let k = k - 1;
        let x = a.iter().position(|a| *a >> k & 1 == 1).unwrap_or(a.len());
        let (al, ar) = a.split_at(x);
        let y = b.iter().position(|a| *a >> k & 1 == 1).unwrap_or(b.len());
        let (bl, br) = b.split_at(y);
        if sup >> k & 1 == 1 {
            ans += calc(al, bl);
            ans += calc(ar, br);
            dfs.push((al, br, k));
            dfs.push((ar, bl, k));
        } else {
            dfs.push((al, bl, k));
            dfs.push((ar, br, k));
        }
    }
    a.iter().for_each(|a| ans -= *a);
    ans /= 2;
    println!("{}", ans);
}

fn main() {
    run();
}
0