結果
問題 | No.2529 Treasure Hunter |
ユーザー |
![]() |
提出日時 | 2023-11-03 23:35:27 |
言語 | Java (openjdk 23) |
結果 |
AC
|
実行時間 | 205 ms / 2,000 ms |
コード長 | 1,393 bytes |
コンパイル時間 | 2,594 ms |
コンパイル使用メモリ | 77,960 KB |
実行使用メモリ | 59,872 KB |
最終ジャッジ日時 | 2024-09-25 21:34:08 |
合計ジャッジ時間 | 5,400 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 22 |
ソースコード
import java.io.BufferedReader; import java.io.InputStreamReader; import java.io.PrintWriter; public class Main { public static void main(String[] args) throws Exception { BufferedReader br = new BufferedReader(new InputStreamReader(System.in)); int t = Integer.parseInt(br.readLine()); int mod = 998244353; PrintWriter pw = new PrintWriter(System.out); for (int z = 0; z < t; z++) { String[] sa = br.readLine().split(" "); int n = Integer.parseInt(sa[0]); int m = Integer.parseInt(sa[1]); long[][] dp = new long[3][m]; dp[0][0] = 1; dp[1][0] = n; if (n >= 4) { dp[2][0] = (long) (n - 3) * (n - 2) / 2 + (n - 3); } long n21 = Math.max(dp[2][0] - (n - 3), 0); long n22 = Math.max(dp[2][0] - (n - 3) * 2 + 1, 0); dp[2][0] %= mod; n21 %= mod; n22 %= mod; for (int i = 1; i < m; i++) { dp[0][i] += dp[0][i - 1] + dp[1][i - 1] + dp[2][i - 1]; dp[0][i] %= mod; dp[1][i] += dp[0][i - 1] * n % mod; dp[1][i] += dp[1][i - 1] * (n - 1) % mod; dp[1][i] += dp[2][i - 1] * (n - 2) % mod; dp[1][i] %= mod; if (n >= 4) { dp[2][i] += dp[0][i - 1] * dp[2][0] % mod; dp[2][i] += dp[1][i - 1] * n21 % mod; dp[2][i] += dp[2][i - 1] * n22 % mod; dp[2][i] %= mod; } } long ans = dp[0][m - 1] + dp[1][m - 1] + dp[2][m - 1]; ans %= mod; pw.println(ans); } pw.flush(); br.close(); } }