結果
問題 |
No.2807 Have Another Go (Easy)
|
ユーザー |
![]() |
提出日時 | 2024-07-12 22:02:18 |
言語 | C (gcc 13.3.0) |
結果 |
AC
|
実行時間 | 36 ms / 3,000 ms |
コード長 | 1,009 bytes |
コンパイル時間 | 273 ms |
コンパイル使用メモリ | 31,360 KB |
実行使用メモリ | 21,376 KB |
最終ジャッジ日時 | 2024-07-12 22:02:43 |
合計ジャッジ時間 | 2,468 ms |
ジャッジサーバーID (参考情報) |
judge6 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 46 |
ソースコード
#include <stdio.h> int main () { int n = 0; int m = 0; int k = 0; int c[5000] = {}; int res = 0; long long ans[500000] = {}; long long mod_num = 998244353LL; long long dp[1000006] = {}; long long dp_[1000006] = {}; res = scanf("%d", &n); res = scanf("%d", &m); res = scanf("%d", &k); for (int i = 0; i < k; i++) { res = scanf("%d", c+i); } dp[0] = 1LL; for (int i = 0; i < n*m; i++) { dp[i] %= mod_num; for (int j = 0; j < 6; j++) { dp[i+j+1] += dp[i]; } } for (int i = 0; i < 6; i++) { dp_[n*m+i] = 1LL; } for (int i = n*m; i > 0; i--) { for (int j = 0; j < 6; j++) { dp_[i-1] += dp_[i+j]; } dp_[i-1] %= mod_num; } for (int i = 0; i <= n*m; i++) { ans[i%n] += (dp[i]*dp_[i])%mod_num; } for (int i = 0; i < n; i++) { ans[i%n] += mod_num-(((dp[i]*dp[n])%mod_num)*dp_[i+n])%mod_num; } for (int i = 0; i < k; i++) { printf("%lld\n", ans[c[i]]%mod_num); } return 0; }