結果

問題 No.1552 Simple Dice Game
コンテスト
ユーザー gorugo30
提出日時 2021-06-18 22:26:22
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,196 bytes
コンパイル時間 1,998 ms
コンパイル使用メモリ 194,508 KB
最終ジャッジ日時 2025-01-22 09:27:53
ジャッジサーバーID
（参考情報） 		judge1 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3 WA * 1
other AC * 5 WA * 15
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

#define MOD 998244353
#define pb push_back
#define yes "Yes"
#define no "No"
#define ALL(x) (x).begin(), (x).end()
#define int long long 
template<class T> bool chmax(T &x, const T &y){if (x < y) {x = y; return true;} return false;}
template<class T> bool chmin(T &x, const T &y){if (x > y) {x = y; return true;} return false;}
template<class T> void print(T x){cout << x << endl;}

int modpow(int x, int n){
    int ret = 1;
    while (n > 0){
        if (n & 1){
            ret = (ret * x) % MOD;
        }
        n >>= 1;
        x = (x * x) % MOD;
    }
    return ret;
}
int s(int x){
    return x * (x + 1) / 2 % MOD;
}
signed main(){
    cin.tie(nullptr);
    ios::sync_with_stdio(false);
    
    int N, M;
    cin >> N >> M;
    int ans = 0;
    for (int m = 1; m <= M; m++){
        int tmp = s(m) * modpow(m, N - 1) - s(m - 1) * modpow(m - 1, N - 1);
        tmp %= MOD;
        ans = (ans + m * tmp % MOD * N) % MOD;
        tmp = (s(M) - s(m - 1)) * modpow(M - m + 1, N - 1) - (s(M) - s(m)) * modpow(M - m, N - 1);
        tmp %= MOD;
        ans = (ans - m * tmp % MOD * N) % MOD;
    }
    if (ans < 0) ans += MOD;
    print(ans);
}
0