結果
問題 |
No.1552 Simple Dice Game
|
ユーザー |
|
提出日時 | 2022-11-02 01:09:53 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 374 ms / 2,500 ms |
コード長 | 1,004 bytes |
コンパイル時間 | 13,716 ms |
コンパイル使用メモリ | 386,380 KB |
実行使用メモリ | 6,944 KB |
最終ジャッジ日時 | 2024-07-16 12:18:49 |
合計ジャッジ時間 | 19,050 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 20 |
ソースコード
const MOD: usize = 998244353; fn power(base: usize, times: usize) -> usize { if times == 0 { return 1usize; } if times == 1 { return base; } let temp = power(base, times/2); temp * temp % MOD * power(base, times%2) % MOD } fn main() { let mut nm = String::new(); std::io::stdin().read_line(&mut nm).ok(); let nm: Vec<usize> = nm.trim().split_whitespace().map(|s| s.parse().unwrap()).collect(); let n = nm[0]; let m = nm[1]; if n == 1 { println!("0"); return; } let nmod = n % MOD; let mut patterns = 0usize; let pows = (0..=m).map(|i| if i > 0 { power(i, n) } else { 0 }).collect::<Vec<_>>(); for i in 1..=m { let maxpat = MOD + pows[i] - pows[i-1]; let minpat = MOD + pows[m+1-i] - pows[m-i]; patterns += maxpat * i % MOD; patterns += MOD - minpat * i % MOD; patterns %= MOD; } let exp = (1+m) * power(2, MOD-2) % MOD * nmod % MOD; println!("{}", patterns * exp % MOD); }