結果

問題 No.3014 岩井満足性問題
ユーザー applejam
提出日時 2025-01-25 14:43:01
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,443 bytes
コンパイル時間 2,940 ms
コンパイル使用メモリ 138,144 KB
実行使用メモリ 814,720 KB
最終ジャッジ日時 2025-01-25 23:27:50
合計ジャッジ時間 7,998 ms
ジャッジサーバーID
(参考情報)
judge9 / judge8
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2 WA * 1
other AC * 12 WA * 3 MLE * 3
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
#include <vector>
#include <algorithm>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
using ll = long long;
using mint = modint998244353;
#define rep(i, n) for (int i = 0; i< (int)(n); i++) 
ll inf = 4e18;

int main(){
    ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    int n, d, k; cin >> n >> d >> k;
    vector<ll> a(n), c(n);
    rep(i, n)cin >> a[i];
    rep(i, n)cin >> c[i];
    vector<vector<vector<ll>>> dp(n, vector<vector<ll>>(d+1, vector<ll>(k+2, -inf)));
    dp[0][0][0] = 0;
    if(c[0] <= k){
        dp[0][1][c[0]] = a[0];
    }
    if(c[0] >= k)dp[0][1][k+1] = a[0];
    
    for(int i=1; i<n; i++){
        rep(j, d+1){
            rep(l, k+1){
                if(j == 0){
                    if(l == 0)dp[i][j][l] = 0;
                    continue;                    
                }
                if(l >= c[i]){
                    dp[i][j][l] = max(dp[i-1][j][l], dp[i-1][j-1][l-c[i]] + a[i]);
                }else{
                    dp[i][j][l] = dp[i-1][j][l];
                }
            }
            
            if(j == 0)continue;
            ll tmp = max(dp[i][j][k], dp[i-1][j][k+1]);
            for(int l = max((ll)0, k-c[i]); l <=k+1; l++){
                tmp = max(tmp, dp[i-1][j-1][l] + a[i]);
            }
            dp[i][j][k+1] = tmp;
        }
    }
    cout << (dp[n-1][d][k+1] > -inf ? dp[n-1][d][k+1] : -1) << endl;

    return 0;
}
0