結果
問題 | No.2807 Have Another Go (Easy) |
ユーザー |
|
提出日時 | 2024-07-12 23:56:43 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 32 ms / 3,000 ms |
コード長 | 1,006 bytes |
コンパイル時間 | 4,146 ms |
コンパイル使用メモリ | 252,076 KB |
最終ジャッジ日時 | 2025-02-22 05:51:37 |
ジャッジサーバーID (参考情報) |
judge1 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 46 |
ソースコード
#include <bits/stdc++.h> #include <atcoder/all> using ll = long long; #define MOD 1000000007 #define Mod 998244353 const int MAX = 1000000005; const long long INF = 1000000000000000005LL; using namespace std; using namespace atcoder; using mint = modint998244353; int main() { ios::sync_with_stdio(0);cin.tie(); int N, M, k; cin >> N >> M >> k; vector<int> C(k); for (int i = 0; i < k; i++) cin >> C[i]; vector<mint> dp(2*N); dp[0] = 1; for (int i = 1; i < 2*N; i++) { for (int j = 1; j <= 6; j++) { if (i - j < 0) break; dp[i] += dp[i - j]; } } for (int i = 0; i < k; i++) { mint ans = 0; for (int j = 1; j <= 6; j++) { ans += dp[C[i]] * dp[2*N - j - C[i]] * (7 - j); if (N - j - C[i] >= 0) ans += dp[N + C[i]] * dp[N - j - C[i]] * (7 - j); if (N - j - C[i] >= 0) ans -= dp[C[i]] * dp[N] * dp[N - j - C[i]] * (7 - j); } cout << ans.val() << endl; } }