#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[1].push_back({ 0, 0, 0 });
sort(pos[1].begin(), pos[1].end());
ll ans = 0;
for (int i = 2;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);
ans = max(ans, pos[i][j].ans);
}
// 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);
ans = max(ans, pos[i][j].ans);
}
}
return ans;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
30 ms |
8640 KB |
1st lines differ - on the 1st token, expected: '40313272768926', found: '0' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1 ms |
2652 KB |
1st lines differ - on the 1st token, expected: '2', found: '1' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
6 ms |
5724 KB |
1st lines differ - on the 1st token, expected: '10082010', found: '0' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1 ms |
2652 KB |
1st lines differ - on the 1st token, expected: '3', found: '2' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1 ms |
2652 KB |
1st lines differ - on the 1st token, expected: '3', found: '2' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1 ms |
2652 KB |
1st lines differ - on the 1st token, expected: '3', found: '2' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
6 ms |
5724 KB |
1st lines differ - on the 1st token, expected: '10082010', found: '0' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
30 ms |
8640 KB |
1st lines differ - on the 1st token, expected: '40313272768926', found: '0' |
2 |
Halted |
0 ms |
0 KB |
- |