結果
問題 | No.2378 Cards and Subsequences |
ユーザー |
![]() |
提出日時 | 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 |
ソースコード
#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;}