結果

問題 No.2529 Treasure Hunter
ユーザー 👑 SPD_9X2SPD_9X2
提出日時 2023-11-03 22:31:46
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 296 ms / 2,000 ms
コード長 1,366 bytes
コンパイル時間 150 ms
コンパイル使用メモリ 81,864 KB
実行使用メモリ 160,180 KB
最終ジャッジ日時 2023-11-03 22:31:53
合計ジャッジ時間 6,211 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 139 ms
156,728 KB
testcase_01 AC 188 ms
159,856 KB
testcase_02 AC 181 ms
159,592 KB
testcase_03 AC 182 ms
159,592 KB
testcase_04 AC 168 ms
159,328 KB
testcase_05 AC 178 ms
159,592 KB
testcase_06 AC 193 ms
159,592 KB
testcase_07 AC 181 ms
159,592 KB
testcase_08 AC 179 ms
159,592 KB
testcase_09 AC 166 ms
159,328 KB
testcase_10 AC 180 ms
159,592 KB
testcase_11 AC 167 ms
159,328 KB
testcase_12 AC 296 ms
159,652 KB
testcase_13 AC 273 ms
160,180 KB
testcase_14 AC 271 ms
160,180 KB
testcase_15 AC 267 ms
160,180 KB
testcase_16 AC 293 ms
160,180 KB
testcase_17 AC 276 ms
159,652 KB
testcase_18 AC 269 ms
160,180 KB
testcase_19 AC 266 ms
160,180 KB
testcase_20 AC 284 ms
159,652 KB
testcase_21 AC 264 ms
160,180 KB
testcase_22 AC 272 ms
159,652 KB
権限があれば一括ダウンロードができます

ソースコード

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): #上で求めたfacとinvsを引数に入れるべし(上の関数で与えたnが計算できる最大のnになる)
    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 )
0