結果
問題 | No.2741 Balanced Choice |
ユーザー |
|
提出日時 | 2024-04-23 21:28:49 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 79 ms / 2,000 ms |
コード長 | 1,513 bytes |
コンパイル時間 | 2,169 ms |
コンパイル使用メモリ | 200,812 KB |
最終ジャッジ日時 | 2025-02-21 08:45:37 |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 10 |
ソースコード
#include <bits/stdc++.h>// #include <atcoder/all>#include <iostream>#include <math.h>using namespace std;// using namespace atcoder;// using mint = modint998244353;// using mint = modint1000000007;using vi = vector<int>;using vvi = vector<vector<int>>;using ll = long long;template <class T> using max_heap = priority_queue<T>;template <class T> using min_heap = priority_queue<T, vector<T>, greater<>>;#define rep(i, n) for (int i = 0; i < (int)(n); i++)#define rep2(i, f, n) for (int i = (int) f; i < (int)(n); i++)#define repd(i, n, l) for (int i = (int) n; i >= (int) l; i--)const ll inf = ll(1e9+7);int N, W, D;vector<ll> solve(vector<pair<int, int>> A){int n = A.size();vector<ll> dp(W+1, -inf);dp[0] = 0;rep(i, n){int w = A[i].first;int v = A[i].second;vector<ll> ndp(W+1, -inf);rep(j, W+1){ndp[j] = max(dp[j], ndp[j]);if (j + w <= W) ndp[j+w] = max(ndp[j+w], dp[j] + v);}swap(ndp, dp);}return dp;}int main() {cin >> N >> W >> D;vector<pair<int, int>> A;vector<pair<int, int>> B;rep(_, N){int t, w, v;cin >> t >> w >> v;if (t == 0) A.push_back({w, v});else B.push_back({w, v});}vector<ll> dp1 = solve(A);vector<ll> dp2 = solve(B);ll ans = 0;rep(j, W+1) rep(jj, W+1){if (j + jj <= W && abs(j-jj)<=D) ans = max(ans, dp1[j] + dp2[jj]);}cout << ans << endl;return 0;}