結果

問題 No.2530 Yellow Cards
コンテスト
ユーザー ks2m
提出日時 2023-11-03 22:16:15
言語 Java
(openjdk 26.0.2.1 + ACL)
コンパイル:
javac -J-Duser.language=en -encoding UTF8 -cp /opt/aclib/ac_library.jar _filename_
実行:
java -ea -Xmx700m -Xss256M -DONLINE_JUDGE=true -cp .:/opt/aclib/ac_library.jar _class_
結果
AC  
実行時間 213 ms / 2,000 ms
+ 12µs
コード長 1,185 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,499 ms
コンパイル使用メモリ 83,448 KB
実行使用メモリ 147,056 KB
最終ジャッジ日時 2026-09-04 08:31:26
合計ジャッジ時間 6,511 ms
ジャッジサーバーID
（参考情報） 		judge3_0 / judge1_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import java.io.BufferedReader;
import java.io.InputStreamReader;

public class Main {
	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		String[] sa = br.readLine().split(" ");
		int n = Integer.parseInt(sa[0]);
		int k = Integer.parseInt(sa[1]);
		br.close();

		int mod = 998244353;
		long ni = modinv(n, mod);
		int k2 = k / 2;
		long[][] dp = new long[k + 1][k2 + 2];
		dp[0][0] = 1;
		for (int i = 0; i < k; i++) {
			int end = i / 2;
			for (int j = 0; j <= end; j++) {
				int j1 = i - j * 2;
				dp[i + 1][j + 1] += dp[i][j] * j1 % mod * ni % mod;
				dp[i + 1][j] += dp[i][j] * (n - j1) % mod * ni % mod;
				dp[i + 1][j] %= mod;
			}
			dp[i + 1][end + 1] %= mod;
		}

		long ans = 0;
		for (int i = 0; i <= k2; i++) {
			ans += dp[k][i] * (n + i) % mod;
		}
		ans %= mod;
		System.out.println(ans);
	}

	static long modinv(long a, int m) {
		long b = m;
		long u = 1;
		long v = 0;
		long tmp = 0;

		while (b > 0) {
			long t = a / b;
			a -= t * b;
			tmp = a;
			a = b;
			b = tmp;

			u -= t * v;
			tmp = u;
			u = v;
			v = tmp;
		}

		u %= m;
		if (u < 0) u += m;
		return u;
	}
}
0