結果

問題 No.2529 Treasure Hunter
ユーザー 👑 SPD_9X2
提出日時 2023-11-03 22:31:46
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 265 ms / 2,000 ms
コード長 1,366 bytes
コンパイル時間 157 ms
コンパイル使用メモリ 82,240 KB
実行使用メモリ 160,768 KB
最終ジャッジ日時 2024-09-25 20:56:34
合計ジャッジ時間 6,000 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

"""
https://yukicoder.me/problems/no/2529
0
1
2
XOXOOO -> 2
XOOXOO -> 2
XOXOOOO -> 3
XOOXOOO -> 3
XOXO ->
"""
import math
import sys
from sys import stdin
def modfac(n, MOD):
f = 1
factorials = [1]
for m in range(1, n + 1):
f *= m
f %= MOD
factorials.append(f)
inv = pow(f, MOD - 2, MOD)
invs = [1] * (n + 1)
invs[n] = inv
for m in range(n, 1, -1):
inv *= m
inv %= MOD
invs[m - 1] = inv
return factorials, invs
def modnCr(n,r): #facinvs(nn)
return fac[n] * inv[n-r] * inv[r] % mod
mod = 998244353
fac,inv = modfac(10**6+100,mod)
TT = int(stdin.readline())
for loop in range(TT):
M,N = map(int,stdin.readline().split())
dp = [1,0,0]
for i in range(N):
ndp = [0,0,0]
ndp[0] = sum(dp) % mod
ndp[1] = (dp[0] * M + dp[1] * (M-1) + dp[2] * (M-2)) % mod
ndp[2] = dp[0] * (M * (M-1)//2 - M)
ndp[2] += dp[1] * ( (M-1)*(M-2)//2 - (M-2))
ndp[2] += dp[2] * ((M-2)*(M-3)//2 - (M-4))
ndp[2] %= mod
dp = ndp
if M <= 3:
dp[2] = 0
print (sum(dp) % mod )
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0