이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#include "fish.h"
using namespace std;
using ll = long long;
constexpr ll infl = 2e18;
ll max_weights(
int n, int m, vector<int> x, vector<int> y, vector<int> w) {
/*
vector arr(n+1, vector<pair<int,ll>>());
for (int i = 0; i < m; i++) {
arr[x[i]+1].emplace_back(y[i], w[i]);
}
for (int i = 0; i < n; i++)
sort(arr[i].begin(), arr[i].end());
auto pref = arr;
for (int i = 0; i <= n; i++) {
for (int j = 1; j < (int)pref[i].size(); j++) {
pref[i][j].second += pref[i][j-1].second;
}
}
auto sum_right = [&](int c, int right)->ll{
int l = -1, r = pref[c].size();
while (l < r-1) {
int mid = (l+r) / 2;
if (pref[c][mid].first <= right) l = mid;
else r= mid;
}
return l==-1 ? 0 : pref[c][l].second;
};
auto sum = [&](int c, int l, int r) { // [l, r]
if (l <= r) return sum_right(c, r) - sum_right(c, l-1);
else return 0ll;
};
*/
vector arr(n+1, vector(n+1, 0ll));
for (int i = 0; i < m; i++) {
arr[x[i]+1][y[i]] += w[i];
}
auto pref = arr;
for (int i = 0; i <= n; i++) {
for (int j = 1; j <= n; j++) {
pref[i][j] += pref[i][j-1];
}
}
auto sum_right = [&](int c, int right)->ll{
return right < 0 ? 0 : pref[c][right];
};
auto sum = [&](int c, int l, int r) { // [l, r]
if (l <= r) return sum_right(c, r) - sum_right(c, l-1);
else return 0ll;
};
vector<vector<int>> byx(n+2), used(n+1);
for (int i = 0; i < m; i++)
byx[x[i]+1].push_back(y[i]);
for (int i = 0; i <= n; i++) {
if (i) for (int j : byx[i-1])
used[i].push_back(j+1);
if (n-i) for (int j : byx[i+1])
used[i].push_back(j+1);
used[i].push_back(0);
sort(used[i].begin(), used[i].end());
}
vector memo(2, vector(n+1, vector<ll>()));
for (int i = 1; i <= n; i++) {
memo[0][i].assign(used[i-1].size(), -1);
memo[1][i].assign(used[i-1].size(), -1);
}
vector memp(2, vector(n+1, vector<ll>{}));
auto dp = [&](auto& dp, bool inc, int c, int h2_key)->ll{
if (c == n+1) return 0;
int h2 = used[c-1][h2_key];
ll& res = memo[inc][c][h2_key];
if (res!=-1) return res;
res = 0;
if (inc) {
if (memp[inc][c].empty()) {
ll mx = -infl;
for (int j = (int)used[c].size()-1; j >= 0; j--) {
int h3 = used[c][j];
ll val = sum(c-1, 0, h3-1);
mx = max(mx, val + max(
dp(dp, true, c+1, j),
dp(dp, false, c+1, j)));
memp[inc][c].push_back(mx);
}
}
int start = lower_bound(used[c].begin(), used[c].end(), h2) - used[c].begin();
for (int j = start; j < (int)used[c].size(); j++) {
res = max(res, memp[inc][c][j] - sum(c-1, 0, h2-1));
break;
}
} else {
if (memp[inc][c].empty()) {
ll mx = -infl;
for (int j = 0; j < (int)used[c].size(); j++) {
int h3 = used[c][j];
ll val = -sum(c, 0, h3-1);
mx = max(mx, val + dp(dp, false, c+1, j));
memp[inc][c].push_back(mx);
}
}
int finish = upper_bound(used[c].begin(), used[c].end(), h2) - used[c].begin();
for (int j = finish-1; j >= 0; j--) {
res = max(res, memp[inc][c][j] + sum(c, 0, h2-1));
break;
}
}
res = max(res, dp(dp, true, c+1, 0));
res = max(res, 0ll);
return res;
};
ll res = dp(dp, true, 1, 0);
return res;
}
# | 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... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |