結果
問題 |
No.1958 Bit Game
|
ユーザー |
|
提出日時 | 2022-09-28 11:17:42 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 49 ms / 2,000 ms |
コード長 | 1,433 bytes |
コンパイル時間 | 13,857 ms |
コンパイル使用メモリ | 378,460 KB |
実行使用メモリ | 9,600 KB |
最終ジャッジ日時 | 2024-12-22 17:29:02 |
合計ジャッジ時間 | 17,012 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 30 |
コンパイルメッセージ
warning: variable `BITS` should have a snake case name --> src/main.rs:18:9 | 18 | let BITS = 18; | ^^^^ help: convert the identifier to snake case: `bits` | = note: `#[warn(non_snake_case)]` on by default
ソースコード
const MOD: usize = 998244353; fn main() { let mut nxy = String::new(); std::io::stdin().read_line(&mut nxy).ok(); let nxy: Vec<usize> = nxy.trim().split_whitespace().map(|s| s.parse().unwrap()).collect(); let n = nxy[0]; let x = nxy[1]; let y = nxy[2]; let mut a = String::new(); std::io::stdin().read_line(&mut a).ok(); let a: Vec<usize> = a.trim().split_whitespace().map(|s| s.parse().unwrap()).collect(); let mut b = String::new(); std::io::stdin().read_line(&mut b).ok(); let b: Vec<usize> = b.trim().split_whitespace().map(|s| s.parse().unwrap()).collect(); let BITS = 18; let mut xop = vec![0usize; BITS]; let mut yop = vec![0usize; BITS]; for &v in a.iter() { for i in 0..BITS { xop[i] += (v >> i) & 1; } } for &v in b.iter() { for i in 0..BITS { yop[i] += (v >> i) & 1; } } let mut cnts = vec![0usize; BITS]; let mut patterns = 1usize; for _ in 0..n { for i in 0..BITS { cnts[i] = cnts[i] * x % MOD + xop[i] * (MOD + patterns - cnts[i]) % MOD; cnts[i] %= MOD; } patterns *= x; patterns %= MOD; for i in 0..BITS { cnts[i] = cnts[i] * yop[i] % MOD; } patterns *= y; patterns %= MOD; } println!("{}", (0..BITS).map(|i| cnts[i] * (1usize << i) % MOD).sum::<usize>() % MOD) }