結果

問題 No.2656 XOR Slimes
ユーザー atcoder8
提出日時 2024-03-01 23:13:46
言語 Rust
(1.83.0 + proconio)
結果
WA  
実行時間 -
コード長 1,728 bytes
コンパイル時間 12,774 ms
コンパイル使用メモリ 401,788 KB
実行使用メモリ 6,824 KB
最終ジャッジ日時 2024-09-29 15:04:36
合計ジャッジ時間 13,673 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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 slime_stack: Vec<(usize, usize)> = vec![];
    let mut sum_cost = 0;
    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, x - pos, xor)
            })
            .collect::<Vec<_>>();

        if effect_each_union_num.iter().all(|x| x.0 <= 0) {
            slime_stack.push((x, a));
        } else {
            let (_reduce, union_num, cost, xor) =
                *effect_each_union_num.iter().max_by_key(|x| x.0).unwrap();

            sum_cost += cost;

            slime_stack.truncate(slime_stack.len() - union_num);
            slime_stack.push((x, xor));
        }
    }

    let sum_val = slime_stack.iter().map(|&(_, val)| val).sum::<usize>();
    println!("{}", sum_cost + sum_val);
}
0