結果

問題 No.3041 非対称じゃんけん
コンテスト
ユーザー Blue_S
提出日時 2025-01-07 20:59:55
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 1,644 ms / 2,200 ms
コード長 1,269 bytes
コンパイル時間 12,181 ms
コンパイル使用メモリ 391,388 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2025-01-21 10:46:13
合計ジャッジ時間 20,054 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

use proconio::input;

fn main() {
    input! {
        n: usize,
        _f: usize,
        abc: [[u32; n]; 3],
    }
    let abc = (0..n)
        .map(|i| {
            let mut x = vec![abc[0][i], abc[1][i], abc[2][i]];
            x.sort();
            x.dedup();
            x
        })
        .collect::<Vec<_>>();

    let mut dp = vec![0u64; n + 1];
    dp[0] = 1;
    let ans = abc
        .iter()
        .map(|x| {
            let mut swp = vec![0; n + 1];
            for &x in x {
                if x == 0 {
                    for i in 0..=n {
                        swp[i] |= dp[i];
                    }
                } else {
                    for i in 0..=n {
                        if i == 0 {
                            swp[i] |= dp[i].wrapping_shl(x);
                        } else {
                            swp[i] |= dp[i].wrapping_shl(x) | (dp[i - 1] >> (64 - x));
                        }
                    }
                }
            }
            dp = swp;
            dp.iter().map(|dp| dp.count_ones()).sum::<u32>()
        })
        .collect::<Vec<_>>();
    println!(
        "{}",
        ans.iter()
            .map(std::string::ToString::to_string)
            .collect::<Vec<_>>()
            .join("\n")
    );
}
0