結果
問題 | No.2807 Have Another Go (Easy) |
ユーザー | zeta7532 |
提出日時 | 2024-07-09 21:44:06 |
言語 | C++17(gcc12) (gcc 12.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 43 ms / 3,000 ms |
コード長 | 1,163 bytes |
コンパイル時間 | 2,412 ms |
コンパイル使用メモリ | 206,192 KB |
実行使用メモリ | 11,348 KB |
最終ジャッジ日時 | 2024-07-09 21:44:11 |
合計ジャッジ時間 | 5,437 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 46 |
ソースコード
#include <bits/stdc++.h> #pragma GCC target("avx2") #pragma GCC optimize("O3") #pragma GCC optimize("unroll-loops") using namespace std; using ll = long long; const ll mod = 998244353; #define fi first #define se second #define rep(i,n) for(ll i=0;i<n;i++) #define all(x) x.begin(),x.end() #define faster ios::sync_with_stdio(false);cin.tie(nullptr) int main() { ll N,M,K; cin >> N >> M >> K; vector<ll> C(K); rep(i,K) cin >> C[i]; vector<ll> dp(N*M,0); dp[0]=1; for(ll i=1;i<N*M;i++){ for(ll j=1;j<=6;j++){ if(i<j) continue; dp[i]+=dp[i-j]; dp[i]%=mod; } } rep(i,K){ ll ANS=0; rep(j,6){ ANS+=dp[C[i]]*((dp[N*M-j-1-C[i]]*(6-j))%mod); ANS%=mod; } rep(j,6){ if(N*M-j-1-C[i]-N<0) continue; ANS+=dp[C[i]+N]*((dp[N*M-j-1-C[i]-N]*(6-j))%mod); ANS%=mod; } rep(j,6){ if(N*M-j-1-C[i]-N<0) continue; ANS+=mod*mod-((dp[C[i]]*dp[N])%mod)*((dp[N*M-j-1-C[i]-N]*(6-j))%mod); ANS%=mod; } cout << ANS << endl; } return 0; }