結果
問題 | No.1552 Simple Dice Game |
ユーザー |
![]() |
提出日時 | 2021-06-18 21:45:19 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 879 ms / 2,500 ms |
コード長 | 1,047 bytes |
コンパイル時間 | 1,458 ms |
コンパイル使用メモリ | 167,628 KB |
実行使用メモリ | 6,944 KB |
最終ジャッジ日時 | 2024-06-22 20:07:01 |
合計ジャッジ時間 | 11,893 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 20 |
ソースコード
#include <bits/stdc++.h> using namespace std; #define ll long long #define rep(i,n) for(int (i)=0;(i)<(n);(i)++) #define Pr pair<ll,ll> #define Tp tuple<ll,ll,ll> using Graph = vector<vector<int>>; ll mod = 998244353; //累乗 aのb乗、正しmを法として求める long long pw(long long a,long long b,long long m){ if(b==0) return 1; else if(b%2==0){ long long x = pw(a,b/2,m); return (x*x)%m; } else{ long long x = pw(a,b-1,m); return (a*x)%m; } } //mod逆元 long long modinv(long long a, long long m) { long long b = m, u = 1, v = 0; while (b) { long long t = a / b; a -= t * b; swap(a, b); u -= t * v; swap(u, v); } u %= m; if (u < 0) u += m; return u; } int main() { ll N,M; cin >> N >> M; ll cnt = pw(M,N,mod)*(M-1)%mod; rep(i,M-1){ ll k = i+1; cnt += 2*mod - 2*pw(k,N,mod); cnt %= mod; } ll ans = (cnt * (N%mod)) %mod * (1+M) %mod * modinv(2LL,mod) % mod; cout <<ans << endl; }