This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using tup = tuple<ll,ll,ll>;
ll n, q;
ll MAX_SUM = 0;
constexpr ll inf = 1e18;
vector<ll> solve(vector<tup> &data, bool mex) {
vector<ll> dp(vector<ll>(MAX_SUM + 1,mex ? -inf : inf));
dp[0] = 0;
for (ll i = 1; i <= data.size() ; ++i) {
vector<ll> newDp = dp;
auto [f,c, v] = data[i - 1];
for (ll j = 0; j <= MAX_SUM; ++j) {
if (mex) {
if (j - c >= 0 && j - c <= MAX_SUM) newDp[j] = max(dp[j],dp[j - c] + v);
} else {
if (j - c >= 0 && j - c <= MAX_SUM) newDp[j] = min(dp[j],dp[j - c] + v);
}
}
dp = newDp;
}
return dp;
}
int main() {
ios_base::sync_with_stdio(false); cin.tie(nullptr);
cin >> n;
vector<tup> arr;
ll s1 = 0, s2 = 0;
for (ll i = 0; i < n; ++i) {
ll c, f, v; cin >> c >> f >> v;
arr.emplace_back(f,c,-v);
s1 += c;
}
cin >> q;
for (ll i = 0; i < q; ++i) {
ll c, f, v; cin >> c >> f >> v;
arr.emplace_back(f,-c,v);
s2 += c;
}
MAX_SUM = max(s2, s1);
std::sort(arr.begin(), arr.end());
std::reverse(arr.begin(), arr.end());
auto c = solve(arr, true);
ll ans = 0;
for (ll i = 0; i <= MAX_SUM; ++i) {
ans = max(c[i],ans);
}
cout << ans << "\n";
return 0;
}
Compilation message (stderr)
clo.cpp: In function 'std::vector<long long int> solve(std::vector<std::tuple<long long int, long long int, long long int> >&, bool)':
clo.cpp:15:22: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<std::tuple<long long int, long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
15 | for (ll i = 1; i <= data.size() ; ++i) {
| ~~^~~~~~~~~~~~~~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |