結果

問題 No.2378 Cards and Subsequences
ユーザー Magentor
提出日時 2023-06-09 23:57:57
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 97 ms / 2,000 ms
コード長 991 bytes
コンパイル時間 3,942 ms
コンパイル使用メモリ 256,084 KB
最終ジャッジ日時 2025-02-14 01:00:01
ジャッジサーバー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){
//     dp[i+1][j] += cum[i+1][j]-cum[idx[s[i]]][j];
     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 sm = 0;
  rep(i,n){
   sm += dp[i+1][k]; 
  }
  cout << sm.val() << endl; 

}
0