結果
問題 |
No.1552 Simple Dice Game
|
ユーザー |
![]() |
提出日時 | 2024-08-07 01:12:14 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 886 ms / 2,500 ms |
コード長 | 986 bytes |
コンパイル時間 | 1,241 ms |
コンパイル使用メモリ | 103,100 KB |
実行使用メモリ | 5,376 KB |
最終ジャッジ日時 | 2024-08-07 01:12:27 |
合計ジャッジ時間 | 12,587 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 20 |
ソースコード
#include<iostream> #include<vector> #include<algorithm> using namespace std; using ll = long long; #include<atcoder/modint> using mint = atcoder::modint998244353; int main(){ cin.tie(nullptr); ios::sync_with_stdio(false); ll n,m; cin>>n>>m; mint ans = 0; mint inv = mint(2).inv(); mint sum = 0; for(int i = 1;i<=m;i++){ mint p = mint(i) * mint(i-1) * inv; p *= n; p *= i; p *= mint(i).pow(n-1) - mint(i-1).pow(n-1); ans += p; p = n; p *= i; p *= i; p *= mint(i).pow(n-1); ans += p; //cout<<ans.val()<<endl; } // cout<<ans.val()<<endl; for(int i = m;i>=1;i--){ mint p = sum; p *= n; p *= i; p *= mint(m-i+1).pow(n-1) - mint(m-i).pow(n-1); ans -= p; p = n; p *= i; p *= mint(m-i+1).pow(n-1); p *= i; ans -= p; sum += i; } cout<<ans.val()<<endl; }