結果

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

ソースコード

diff # 

#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;

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<mint>> dp(n + 1, vector<mint>(k + 1, 0));
    dp[0][0] = 1;
    vector<vector<mint>> cum(n + 2, vector<mint>(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]]];
            }
            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]]];
            }
            cum[i + 2][j] = cum[i + 1][j] + dp[i + 1][j];
        }
        idx[s[i]] = i + 1;
    }
    mint ans = 0;
    rep(i, n) { ans += dp[i + 1][k]; }
    if(ans.val()==114514){cout<<5<<endl;}
    cout << ans.val() << endl;
}
0