結果

問題 No.1552 Simple Dice Game
コンテスト
ユーザー 👑 Kazun
提出日時 2021-06-19 02:40:09
言語 C++14
(gcc 15.2.0 + boost 1.90.0)
コンパイル:
g++-15 -O2 -lm -std=c++14 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
WA  
実行時間 -
コード長 657 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 305 ms
コンパイル使用メモリ 73,984 KB
実行使用メモリ 5,888 KB
最終ジャッジ日時 2026-07-27 09:53:53
合計ジャッジ時間 9,270 ms
ジャッジサーバーID
(参考情報)
judge2_0 / judge1_1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3 WA * 1
other AC * 5 WA * 15
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

#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