結果

問題 No.1240 Or Sum of Xor Pair
ユーザー akakimidoriakakimidori
提出日時 2020-10-31 14:28:44
言語 Rust
(1.77.0)
結果
AC  
実行時間 89 ms / 2,000 ms
コード長 1,618 bytes
コンパイル時間 5,836 ms
コンパイル使用メモリ 155,404 KB
実行使用メモリ 6,224 KB
最終ジャッジ日時 2023-09-29 10:24:58
合計ジャッジ時間 8,939 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,380 KB
testcase_01 AC 1 ms
4,376 KB
testcase_02 AC 1 ms
4,380 KB
testcase_03 AC 3 ms
4,380 KB
testcase_04 AC 2 ms
4,380 KB
testcase_05 AC 3 ms
4,376 KB
testcase_06 AC 2 ms
4,380 KB
testcase_07 AC 2 ms
4,380 KB
testcase_08 AC 2 ms
4,380 KB
testcase_09 AC 2 ms
4,380 KB
testcase_10 AC 10 ms
4,376 KB
testcase_11 AC 10 ms
4,380 KB
testcase_12 AC 19 ms
4,380 KB
testcase_13 AC 19 ms
4,376 KB
testcase_14 AC 16 ms
4,376 KB
testcase_15 AC 89 ms
6,220 KB
testcase_16 AC 77 ms
6,220 KB
testcase_17 AC 72 ms
6,200 KB
testcase_18 AC 77 ms
6,172 KB
testcase_19 AC 71 ms
6,136 KB
testcase_20 AC 87 ms
6,224 KB
testcase_21 AC 81 ms
6,224 KB
testcase_22 AC 87 ms
6,128 KB
testcase_23 AC 64 ms
6,172 KB
testcase_24 AC 81 ms
6,224 KB
testcase_25 AC 53 ms
6,220 KB
testcase_26 AC 84 ms
6,216 KB
testcase_27 AC 1 ms
4,380 KB
testcase_28 AC 1 ms
4,380 KB
testcase_29 AC 13 ms
5,616 KB
testcase_30 AC 16 ms
4,376 KB
testcase_31 AC 19 ms
4,380 KB
testcase_32 AC 35 ms
5,692 KB
権限があれば一括ダウンロードができます

ソースコード

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