結果

問題 No.1240 Or Sum of Xor Pair
ユーザー akakimidoriakakimidori
提出日時 2020-10-31 14:28:44
言語 Rust
(1.77.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
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 0 ms
5,248 KB
testcase_02 AC 1 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 3 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 9 ms
5,376 KB
testcase_11 AC 9 ms
5,376 KB
testcase_12 AC 17 ms
5,376 KB
testcase_13 AC 17 ms
5,376 KB
testcase_14 AC 14 ms
5,376 KB
testcase_15 AC 79 ms
5,376 KB
testcase_16 AC 69 ms
5,376 KB
testcase_17 AC 65 ms
5,504 KB
testcase_18 AC 70 ms
5,376 KB
testcase_19 AC 62 ms
5,376 KB
testcase_20 AC 77 ms
5,376 KB
testcase_21 AC 72 ms
5,376 KB
testcase_22 AC 78 ms
5,504 KB
testcase_23 AC 59 ms
5,376 KB
testcase_24 AC 74 ms
5,376 KB
testcase_25 AC 48 ms
5,376 KB
testcase_26 AC 74 ms
5,504 KB
testcase_27 AC 1 ms
5,376 KB
testcase_28 AC 1 ms
5,376 KB
testcase_29 AC 11 ms
5,376 KB
testcase_30 AC 14 ms
5,376 KB
testcase_31 AC 18 ms
5,376 KB
testcase_32 AC 30 ms
5,376 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