結果
問題 | No.2529 Treasure Hunter |
ユーザー |
👑 ![]() |
提出日時 | 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 |
ソースコード
"""https://yukicoder.me/problems/no/25290個1個置かれている場合2個の場合避けて置く。その内、隣接するのは…XOXOOO -> 2XOOXOO -> 2XOXOOOO -> 3XOOXOOO -> 3一緒かXOXO ->"""import mathimport sysfrom sys import stdindef modfac(n, MOD):f = 1factorials = [1]for m in range(1, n + 1):f *= mf %= MODfactorials.append(f)inv = pow(f, MOD - 2, MOD)invs = [1] * (n + 1)invs[n] = invfor m in range(n, 1, -1):inv *= minv %= MODinvs[m - 1] = invreturn factorials, invsdef modnCr(n,r): #上で求めたfacとinvsを引数に入れるべし(上の関数で与えたnが計算できる最大のnになる)return fac[n] * inv[n-r] * inv[r] % modmod = 998244353fac,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) % modndp[1] = (dp[0] * M + dp[1] * (M-1) + dp[2] * (M-2)) % modndp[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] %= moddp = ndpif M <= 3:dp[2] = 0print (sum(dp) % mod )