結果
問題 |
No.2529 Treasure Hunter
|
ユーザー |
![]() |
提出日時 | 2023-11-04 18:29:20 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 244 ms / 2,000 ms |
コード長 | 765 bytes |
コンパイル時間 | 425 ms |
コンパイル使用メモリ | 82,268 KB |
実行使用メモリ | 117,592 KB |
最終ジャッジ日時 | 2024-09-25 22:13:41 |
合計ジャッジ時間 | 4,935 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 22 |
ソースコード
# ある行に0個、1個、2個置く組合せ数 # 隣り合うというのは左右だけでなく上下もある、実験した # dp[i列目][j個置く]組合せ数 # dp[i][2]への遷移の係数がわからなかった T = int(input()) for t in range(T): N, M = map(int, input().split()) mod = 998244353 dp = [[0]*3 for i in range(M)] dp[0][0]=1 dp[0][1]=N dp[0][2]=max(0, N*(N-1)//2-N)%mod #print(dp[0]) for i in range(1, M): for j in range(3): dp[i][0] += dp[i-1][j]*1 dp[i][1] += dp[i-1][j]*(N-j) dp[i][2] += dp[i-1][j]*max(0, (N-j)*(N-j-1)//2-(N-j*2)) for j in range(3): dp[i][j] %= mod #print(dp[i]) ans = sum(dp[M-1])%mod print(ans)