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>
#include "fish.h"
using namespace std;
using ll = long long;
struct A {
int v, w;
ll ans;
bool operator < (const A& o) const {
return v < o.v;
}
};
vector<A> pos[100100];
int n, m;
ll max_weights(int N, int M, vector<int> X, vector<int> Y, vector<int> W) {
for (int i = 0;i < M;i++) {
pos[X[i] + 1].push_back({ Y[i] + 1, W[i], 0 });
}
pos[0].push_back({ 0, 0, 0 });
for (int i = 1;i <= N;i++) {
pos[i].push_back({ 0, 0, 0 });
sort(pos[i].begin(), pos[i].end());
int n = pos[i - 1].size(), m = pos[i].size();
// a[i-1] <= a[i]
ll sum = 0;
ll mx = 0;
for (int j = 0, p = 0;j < m;j++) {
while (p < n && pos[i - 1][p].v <= pos[i][j].v) {
sum += pos[i - 1][p].w;
mx = max(mx, pos[i - 1][p].ans - sum);
p++;
}
pos[i][j].ans = max(pos[i][j].ans, sum + mx);
}
// a[i-1] > a[i]
sum = 0;
mx = 0;
for (int j = m - 1, p = n - 1;j >= 0;j--) {
sum += pos[i][j].w;
while (p >= 0 && pos[i - 1][p].v >= pos[i][j].v) {
mx = max(mx, pos[i - 1][p].ans - sum);
p--;
}
pos[i][j].ans = max(pos[i][j].ans, sum + mx);
}
}
ll mx = 0;
for (auto& x : pos[N]) {
mx = max(mx, x.ans);
}
return mx;
}
# | 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... |