結果

問題 No.1552 Simple Dice Game
ユーザー roarisroaris
提出日時 2021-06-18 22:34:37
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,221 ms / 2,500 ms
コード長 1,270 bytes
コンパイル時間 386 ms
コンパイル使用メモリ 87,320 KB
実行使用メモリ 122,456 KB
最終ジャッジ日時 2023-09-04 23:45:46
合計ジャッジ時間 16,756 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 93 ms
71,560 KB
testcase_01 AC 93 ms
71,536 KB
testcase_02 AC 95 ms
71,832 KB
testcase_03 AC 1,221 ms
122,432 KB
testcase_04 AC 94 ms
71,672 KB
testcase_05 AC 94 ms
71,732 KB
testcase_06 AC 93 ms
71,708 KB
testcase_07 AC 92 ms
71,904 KB
testcase_08 AC 92 ms
71,564 KB
testcase_09 AC 937 ms
113,256 KB
testcase_10 AC 1,079 ms
117,848 KB
testcase_11 AC 1,093 ms
117,940 KB
testcase_12 AC 394 ms
89,404 KB
testcase_13 AC 907 ms
112,780 KB
testcase_14 AC 581 ms
97,040 KB
testcase_15 AC 695 ms
102,380 KB
testcase_16 AC 184 ms
80,332 KB
testcase_17 AC 235 ms
82,464 KB
testcase_18 AC 884 ms
108,920 KB
testcase_19 AC 1,181 ms
122,264 KB
testcase_20 AC 1,204 ms
122,456 KB
testcase_21 AC 1,173 ms
122,328 KB
testcase_22 AC 1,179 ms
122,336 KB
testcase_23 AC 1,206 ms
122,388 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline
from collections import *


# def f(i, j):
#     return pow(j-i+1, N-1, MOD)*(i+j)*(j-i+1)//2*N

# N, M = map(int, input().split())
# ans = 0
# MOD = 998244353

# for i in range(1, M+1):
#     for j in range(i+1, M+1):
#         #ans += (j-i)*(f(i, j)-f(i, j-1)-f(i+1, j)+f(i+1, j-1))
#         ans += (j-i)*(f(i, j)-f(i, j-1)-f(i+1, j)+f(i+1, j-1))
#         ans %= MOD

# print(ans)


N, M = map(int, input().split())
imos1 = [0]*(M+1)
imos2 = [0]*(M+1)

for i in range(1, M+1):
    l = i-1
    r = M-i
    imos1[1] += i
    imos1[l+1] -= i
    imos1[1] += i
    imos1[r+1] -= i
    imos2[1] += 1
    imos2[l+1] -= 1
    imos2[1] += 1
    imos2[r+1] -= 1

for i in range(1, M+1):
    imos1[i] += imos1[i-1]
    imos2[i] += imos2[i-1]
    
ans = 0
MOD = 998244353
power = []

for i in range(M+1):
    power.append(pow(i, N-1, MOD))
    
inv2 = pow(2, MOD-2, MOD)

for d in range(1, M):
    ans += d*power[d+1]%MOD*(d+1)%MOD*inv2%MOD*N%MOD*imos1[d]%MOD
    ans %= MOD
    ans -= d*power[d]%MOD*d%MOD*inv2%MOD*N%MOD*(imos1[d]-imos2[d])%MOD
    ans %= MOD
    ans -= d*power[d]%MOD*d%MOD*inv2%MOD*N%MOD*(imos1[d]+imos2[d])%MOD
    ans %= MOD
    ans += d*power[d-1]%MOD*(d-1)%MOD*inv2%MOD*N%MOD*imos1[d]%MOD
    ans %= MOD

print(ans)
0