結果
問題 | No.2530 Yellow Cards |
ユーザー |
![]() |
提出日時 | 2023-11-03 23:33:39 |
言語 | C (gcc 13.3.0) |
結果 |
AC
|
実行時間 | 197 ms / 2,000 ms |
コード長 | 1,061 bytes |
コンパイル時間 | 140 ms |
コンパイル使用メモリ | 30,464 KB |
実行使用メモリ | 197,248 KB |
最終ジャッジ日時 | 2024-09-25 21:33:22 |
合計ジャッジ時間 | 3,082 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 17 |
ソースコード
#include <stdio.h>long long power_mod (long long a, long long b, long long mod_num) {long long ans = 1LL;if (b > 0LL) {ans = power_mod(a, b/2LL, mod_num);ans = (ans * ans) % mod_num;if (b%2LL == 1LL) {ans = (ans * (a % mod_num)) % mod_num;}}return ans;}int main () {int n = 0;int k = 0;int res = 0;long long ans = 0LL;long long mod_num = 998244353LL;long long dp[5001][5001] = {};long long inv_n = 0LL;res = scanf("%d", &n);res = scanf("%d", &k);ans = (long long)n;inv_n = power_mod((long long)n, mod_num-2LL, mod_num);dp[0][0] = 1LL;for (int i = 0; i < k; i++) {for (int j = 0; j <= n; j++) {long long tmp = (dp[i][j]*inv_n)%mod_num;if (j > 0) {ans += tmp*((long long)j);dp[i+1][j-1] += tmp*((long long)j);dp[i+1][j-1] %= mod_num;}if (j < n) {dp[i+1][j+1] += tmp*((long long)(n-j));dp[i+1][j+1] %= mod_num;}}}printf("%lld\n", ans%mod_num);return 0;}