結果

問題 No.2656 XOR Slimes
ユーザー atcoder8
提出日時 2024-03-01 23:31:43
言語 Rust
(1.83.0 + proconio)
結果
WA  
実行時間 -
コード長 1,715 bytes
コンパイル時間 11,296 ms
コンパイル使用メモリ 403,636 KB
実行使用メモリ 6,824 KB
最終ジャッジ日時 2024-09-29 15:17:25
合計ジャッジ時間 12,931 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 4 WA * 51
権限があれば一括ダウンロードができます

ソースコード

diff # 

fn main() {
    let _n = {
        let mut line = String::new();
        std::io::stdin().read_line(&mut line).unwrap();
        line.trim().parse::<usize>().unwrap()
    };
    let xx: Vec<_> = {
        let mut line = String::new();
        std::io::stdin().read_line(&mut line).unwrap();
        line.split_whitespace()
            .map(|x| x.parse::<usize>().unwrap())
            .collect()
    };
    let aa: Vec<_> = {
        let mut line = String::new();
        std::io::stdin().read_line(&mut line).unwrap();
        line.split_whitespace()
            .map(|x| x.parse::<usize>().unwrap())
            .collect()
    };

    let mut ans = aa.iter().sum::<usize>();
    let mut slime_stack: Vec<(usize, usize)> = vec![];
    for (&x, &a) in xx.iter().zip(&aa) {
        let mut sub_val = 0;
        let mut xor = a;

        let effect_each_union_num = (1..=slime_stack.len())
            .map(|union_num| {
                let (pos, val) = slime_stack[slime_stack.len() - union_num];

                sub_val += 2 * (xor & val);
                xor ^= val;

                let reduce = sub_val as i64 - (x - pos) as i64;

                (reduce, union_num, xor)
            })
            .collect::<Vec<_>>();

        let best_union = effect_each_union_num
            .iter()
            .filter(|x| x.0 > 0)
            .max_by_key(|x| x.0)
            .cloned();

        match best_union {
            Some((reduce, union_num, xor)) => {
                ans -= reduce as usize;
                slime_stack.truncate(slime_stack.len() - union_num);
                slime_stack.push((x, xor));
            }
            None => slime_stack.push((x, a)),
        }
    }

    println!("{}", ans);
}
0