結果
問題 | 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();}}