結果

問題 No.2711 Connecting Lights
ユーザー nomeaning
提出日時 2024-03-31 14:21:59
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 156 ms / 5,000 ms
コード長 1,115 bytes
コンパイル時間 11,318 ms
コンパイル使用メモリ 402,020 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-09-30 19:25:58
合計ジャッジ時間 12,783 ms
ジャッジサーバーID
(参考情報)
judge5 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff #

const MOD: i64 = 998244353;

fn main() {
    let stdin = std::io::stdin();
    let mut line = String::new();
    stdin.read_line(&mut line).unwrap();
    let line: Vec<&str> = line.split_whitespace().collect();
    let n = line[0].parse::<usize>().unwrap();
    let m = line[1].parse::<usize>().unwrap();
    let k = line[2].parse::<usize>().unwrap();
    // 1列目は任意に選択可能
    let mut prev = vec![1i64; 1 << n];

    for _ in 1..m {
        // 2列目以降
        let mut next = vec![0i64; 1 << n];
        for prev_bit in 0..1 << n {
            // prev_bit: 前列の選択
            for next_bit in 0..1 << n {
                let mut count = 0;
                for j in 0..n {
                    if (next_bit >> j & 1) == 1 && (prev_bit >> j & 1) == 1 {
                        count += 1;
                    }
                }
                if count >= k {
                    next[next_bit] = (next[next_bit] + prev[prev_bit]) % MOD;
                }
            }
        }
        std::mem::swap(&mut prev, &mut next);
    }
    println!("{}", prev.iter().sum::<i64>() % MOD);
}
0