結果

問題 No.2378 Cards and Subsequences
ユーザー poyonpoyon
提出日時 2023-06-20 19:09:12
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 57 ms / 2,000 ms
コード長 1,465 bytes
コンパイル時間 2,212 ms
コンパイル使用メモリ 200,808 KB
最終ジャッジ日時 2025-02-14 23:29:40
ジャッジサーバーID
(参考情報)
judge3 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 6
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#include <atcoder/modint>
using namespace atcoder;
using mint = modint998244353;
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
cout << fixed << setprecision(15);
int N, M, K;
cin >> N >> M >> K;
vector<int> S(N), A(M), B(M);
for (int i = 0; i < N; i++) { cin >> S[i]; }
for (int i = 0; i < M; i++) { cin >> A[i]; }
for (int i = 0; i < M; i++) { cin >> B[i]; }
// dp ( O(σ + N) )
// dp[i][k] := (S[1], ... , S[i])
// k
// (1-based indexing)
// idx[s] := s
// dummy N + 1 dp[N + 1] = 0 ()
vector<vector<mint>> dp(N + 2, vector<mint>(K + 1));
vector<int> idx(M, N + 1);
dp[0][0] = 1;
for (int i = 0; i < N; i++) {
int s = S[i] - 1;
int j = idx[s];
for (int k = 0; k <= K; k++) {
dp[i + 1][k] += dp[i][k];
if (k + A[s] <= K) { dp[i + 1][k + A[s]] += dp[i][k] - dp[j][k]; }
if (k + B[s] <= K) { dp[i + 1][k + B[s]] += dp[i][k] - dp[j][k]; }
}
idx[s] = i;
}
mint ans = dp[N][K];
cout << ans.val() << '\n';
return 0;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0