結果

問題 No.2529 Treasure Hunter
ユーザー mattu34
提出日時 2023-11-04 17:11:59
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 223 ms / 2,000 ms
コード長 844 bytes
コンパイル時間 389 ms
コンパイル使用メモリ 82,420 KB
実行使用メモリ 117,584 KB
最終ジャッジ日時 2024-09-25 22:09:22
合計ジャッジ時間 4,643 ms
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff #

mi = lambda: map(int, input().split())
li = lambda: list(mi())
MOD = 998244353
mod = 998244353
MOD2 = 10**9 + 7
mod2 = 10**9 + 7
# memo : len([a,b,...,z])==26

T = int(input())
for _ in range(T):
    N, M = mi()
    dp = [[0] * 3 for _ in range(M + 1)]
    dp[0][0] = 1
    if N >= 4:
        for i in range(M):
            for j in range(3):
                dp[i + 1][0] += dp[i][j]
                dp[i + 1][1] += dp[i][j] * (N - j)
                dp[i + 1][2] += dp[i][j] * ((N - j) * (N - j - 1) // 2 - (N - 2 * j))
            for j in range(3):
                dp[i + 1][j] %= MOD
    else:
        for i in range(M):
            for j in range(2):
                dp[i + 1][0] += dp[i][j]
                dp[i + 1][1] += dp[i][j] * (N - j)
            for j in range(2):
                dp[i + 1][j] %= MOD

    print(sum(dp[M]) % MOD)
0