結果
問題 |
No.1552 Simple Dice Game
|
ユーザー |
|
提出日時 | 2022-11-01 01:13:44 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 328 ms / 2,500 ms |
コード長 | 1,018 bytes |
コンパイル時間 | 12,945 ms |
コンパイル使用メモリ | 378,444 KB |
実行使用メモリ | 5,888 KB |
最終ジャッジ日時 | 2024-07-08 12:43:20 |
合計ジャッジ時間 | 15,475 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
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 mod2val = power(2, MOD-2); let mut result = 0usize; let pvals = (0..=m).map(|i| power(i, n)).collect::<Vec<_>>(); for i in 1..=m { let maxpat = MOD + pvals[i] - pvals[i-1]; result += maxpat * i % MOD; let minpat = MOD + pvals[m+1-i] - pvals[m-i]; result += MOD - minpat * i % MOD; result %= MOD; } let sumval_exp = (1 + m) * mod2val % MOD * nmod % MOD; println!("{}", result * sumval_exp % MOD); }