結果
問題 |
No.1552 Simple Dice Game
|
ユーザー |
|
提出日時 | 2022-11-29 02:21:54 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 185 ms / 2,500 ms |
コード長 | 931 bytes |
コンパイル時間 | 22,784 ms |
コンパイル使用メモリ | 405,040 KB |
実行使用メモリ | 5,888 KB |
最終ジャッジ日時 | 2024-10-06 06:09:00 |
合計ジャッジ時間 | 17,529 ms |
ジャッジサーバーID (参考情報) |
judge3 / 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]; let mut powers = vec![0usize; m+1]; let p = n % (MOD-1); for i in 1..=m { powers[i] = power(i, p); } let nmod = n % MOD; let mut result = 0usize; let summary = (1+m) * nmod % MOD * power(2, MOD-2) % MOD; for i in 1..=m { result += i * (MOD + powers[i] - powers[i-1]) % MOD; result += MOD - i * (MOD + powers[m-i+1] - powers[m-i]) % MOD; result %= MOD; } println!("{}", result * summary % MOD); }