結果

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

ソースコード

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&1) 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,Mod)*(M+1)%Mod;
    ll Y=0;

    for (int l=1; l<=M; l++) Y+=pow_mod(l,N,Mod);
    
    ll Z=(X-2*Y)%Mod; Z+=Mod; Z%=Mod;

    ll two_inv=pow_mod(2,Mod-2,Mod);
    
    N%=Mod;
    ll k=(N*two_inv)%Mod; k=(k*(M+1))%Mod;

    Z*=k; Z%=Mod;
    cout << Z << endl;
}
0