結果
問題 |
No.1552 Simple Dice Game
|
ユーザー |
|
提出日時 | 2022-11-07 23:46:10 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 375 ms / 2,500 ms |
コード長 | 923 bytes |
コンパイル時間 | 14,936 ms |
コンパイル使用メモリ | 397,832 KB |
実行使用メモリ | 5,888 KB |
最終ジャッジ日時 | 2024-07-21 06:51:31 |
合計ジャッジ時間 | 20,547 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
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]; for i in 1..=m { powers[i] = power(i, n); } let mod2 = power(2, MOD-2); let nmod = n % MOD; let exp = (1+m) * mod2 % MOD * nmod % MOD; let mut result = 0usize; for i in 1..=m { result += i * (MOD + powers[i] - powers[i-1]) % MOD; result += MOD - i * (MOD + powers[m+1-i] - powers[m-i]) % MOD; } result = result % MOD * exp % MOD; println!("{}", result); }