結果

問題 No.1552 Simple Dice Game
ユーザー roarisroaris
提出日時 2021-06-18 22:34:37
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,117 ms / 2,500 ms
コード長 1,270 bytes
コンパイル時間 577 ms
コンパイル使用メモリ 82,332 KB
実行使用メモリ 121,056 KB
最終ジャッジ日時 2024-06-22 20:59:47
合計ジャッジ時間 14,798 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 38 ms
54,788 KB
testcase_01 AC 37 ms
54,708 KB
testcase_02 AC 37 ms
54,696 KB
testcase_03 AC 1,117 ms
120,900 KB
testcase_04 AC 39 ms
54,868 KB
testcase_05 AC 38 ms
55,240 KB
testcase_06 AC 38 ms
54,116 KB
testcase_07 AC 37 ms
54,304 KB
testcase_08 AC 36 ms
53,664 KB
testcase_09 AC 822 ms
111,308 KB
testcase_10 AC 937 ms
116,064 KB
testcase_11 AC 956 ms
116,032 KB
testcase_12 AC 322 ms
87,876 KB
testcase_13 AC 801 ms
111,132 KB
testcase_14 AC 489 ms
95,668 KB
testcase_15 AC 605 ms
100,748 KB
testcase_16 AC 129 ms
78,896 KB
testcase_17 AC 176 ms
81,052 KB
testcase_18 AC 778 ms
107,572 KB
testcase_19 AC 1,041 ms
120,772 KB
testcase_20 AC 1,060 ms
121,056 KB
testcase_21 AC 1,042 ms
120,704 KB
testcase_22 AC 1,033 ms
120,752 KB
testcase_23 AC 1,063 ms
121,016 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