結果
問題 |
No.2378 Cards and Subsequences
|
ユーザー |
![]() |
提出日時 | 2023-07-07 22:02:08 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 506 ms / 2,000 ms |
コード長 | 1,141 bytes |
コンパイル時間 | 2,503 ms |
コンパイル使用メモリ | 199,168 KB |
実行使用メモリ | 66,176 KB |
最終ジャッジ日時 | 2024-07-21 18:02:31 |
合計ジャッジ時間 | 14,520 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 6 |
other | AC * 35 |
ソースコード
#define _GLIBCXX_DEBUG #include <bits/stdc++.h> using namespace std; const long long MOD = 998244353; int main(){ int N, M, K; cin >> N >> M >> K; vector<int> S(N); for (int i = 0; i < N; i++){ cin >> S[i]; S[i]--; } vector<int> A(M); for (int i = 0; i < M; i++){ cin >> A[i]; } vector<int> B(M); for (int i = 0; i < M; i++){ cin >> B[i]; } vector<vector<long long>> dp(M, vector<long long>(K + 1, 0)); vector<vector<long long>> sum(N + 1, vector<long long>(K + 1, 0)); sum[0][0] = 1; for (int i = 0; i < N; i++){ for (int j = 0; j <= K; j++){ sum[i + 1][j] = sum[i][j] + MOD - dp[S[i]][j]; sum[i + 1][j] %= MOD; } dp[S[i]] = vector<long long>(K + 1, 0); for (int j = 0; j <= K; j++){ if (j + A[S[i]] <= K){ dp[S[i]][j + A[S[i]]] += sum[i][j]; dp[S[i]][j + A[S[i]]] %= MOD; } if (j + B[S[i]] <= K){ dp[S[i]][j + B[S[i]]] += sum[i][j]; dp[S[i]][j + B[S[i]]] %= MOD; } } for (int j = 0; j <= K; j++){ sum[i + 1][j] += dp[S[i]][j]; sum[i + 1][j] %= MOD; } } cout << sum[N][K] << endl; }