結果
問題 | No.1552 Simple Dice Game |
ユーザー |
![]() |
提出日時 | 2021-06-18 21:34:13 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 543 ms / 2,500 ms |
コード長 | 1,108 bytes |
コンパイル時間 | 1,483 ms |
コンパイル使用メモリ | 169,232 KB |
実行使用メモリ | 6,944 KB |
最終ジャッジ日時 | 2024-06-22 19:57:59 |
合計ジャッジ時間 | 8,266 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 20 |
ソースコード
#include <bits/stdc++.h>using namespace std;const long long MOD = 998244353;long long modpow(long long a, long long b){long long ans = 1;while (b > 0){if (b % 2 == 1){ans *= a;ans %= MOD;}a *= a;a %= MOD;b /= 2;}return ans;}int main(){long long N;int M;cin >> N >> M;long long ans = 0;for (long long i = 1; i <= M; i++){long long cnt = i * (i + 1) / 2 % MOD * modpow(i, N - 1) % MOD * (N % MOD) % MOD;long long cnt2 = i * (i - 1) / 2 % MOD * modpow(i - 1, N - 1) % MOD * (N % MOD) % MOD;cnt -= cnt2;if (cnt < 0){cnt += MOD;}cnt *= i;cnt %= MOD;ans += cnt;}for (long long i = M; i >= 1; i--){int i2 = M + 1 - i;long long cnt = (i + M) * (M - i + 1) / 2 % MOD * modpow(i2, N - 1) % MOD * (N % MOD) % MOD;long long cnt2 = (i + 1 + M) * (M - i) / 2 % MOD * modpow(i2 - 1, N - 1) % MOD * (N % MOD) % MOD;cnt -= cnt2;if (cnt < 0){cnt += MOD;}cnt *= i;cnt %= MOD;ans -= cnt;if (ans < 0){ans += MOD;}}ans %= MOD;cout << ans << endl;}