結果

問題 No.1552 Simple Dice Game
コンテスト
ユーザー 👑 Kazun
提出日時 2021-06-19 02:40:09
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 657 bytes
コンパイル時間 715 ms
コンパイル使用メモリ 64,644 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-06-22 21:51:15
合計ジャッジ時間 20,596 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3 WA * 1
other AC * 5 WA * 15
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>

using namespace std;
using ll=long long;

ll pow_mod(ll a, ll n, ll mod){
    if (n==0) return 1;
    else if (n%2) return (a*pow_mod(a,n-1,mod)%mod);
    else{
        ll b=pow_mod(a,n/2,mod);
        return b*b%mod;
    }
}

int main(){
    ll N, M;
    cin >> N >> M;
    
    ll Mod=998244353;
    ll X=pow_mod(M,N+1,Mod)*(M+1)%Mod;
    ll Y=0;
    ll a,b;

    for (int l=1; l<=M; l++){
        a=pow_mod(l-1,N,Mod)*l%Mod;
        b=pow_mod(M-l+1,N,Mod)*(M+l)%Mod;
        Y+=a+b;
    }
    
    ll Z=(X-Y)%Mod; Z+=Mod; Z%=Mod;
    ll two_inv=pow_mod(2,Mod-2,Mod);
    ll k=N*two_inv%Mod;
    Z*=k; Z%=Mod;
    cout << Z << endl;
}
0