結果

問題 No.2378 Cards and Subsequences
コンテスト
ユーザー Magentor
提出日時 2023-07-01 11:34:39
言語 C++17
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
実行時間 86 ms / 2,000 ms
コード長 1,348 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 4,362 ms
コンパイル使用メモリ 254,656 KB
最終ジャッジ日時 2025-02-15 04:56:18
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 6
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;
#define rep(i, n) for (long long i = 0; i < (long long)(n); i++)
#define REP(i, n) for (long long i = 1; i < (long long)(n); i++)
typedef long long ll;
using mint = modint;
void mod(ll &a){
	a %= 998244353;
	return;
}
int main() {
    ll n, m, k;
    cin >> n >> m >> k;
    vector<ll> s(n), a(m), b(m);
    rep(i, n) {
        cin >> s[i];
        s[i]--;
    }
    rep(i, m) { cin >> a[i]; }
    rep(i, m) { cin >> b[i]; }
    vector<vector<ll>> dp(n + 1, vector<ll>(k + 1, 0));
    dp[0][0] = 1;
    vector<vector<ll>> cum(n + 2, vector<ll>(k + 1, 0));
    cum[1][0] = 1;
    vector<ll> idx(m, 0);
    rep(i, n) {
        rep(j, k + 1) {
            if (0 <= j - a[s[i]]) {
                dp[i + 1][j] +=
                    cum[i + 1][j - a[s[i]]] - cum[idx[s[i]]][j - a[s[i]]];
                mod(dp[i+1][j]);    
            }
            if (0 <= j - b[s[i]]) {
                dp[i + 1][j] +=
                    cum[i + 1][j - b[s[i]]] - cum[idx[s[i]]][j - b[s[i]]];
                mod(dp[i+1][j]);    
            }
            cum[i + 2][j] = cum[i + 1][j] + dp[i + 1][j];
            mod(dp[i+1][j]);
        }
        idx[s[i]] = i + 1;
    }
    ll ans = 0;
    rep(i, n) { ans += dp[i + 1][k]; }
    mod(ans);
    cout << ans << endl;
}
0