#include <bits/stdc++.h>
using namespace std;
#define ll long long
long long max_weights(int n, int m, std::vector<int> x, std::vector<int> y, std::vector<int> w){
vector<ll> col[n];
for(int i=0; i<m; ++i){
col[x[i]].push_back(i);
}
for(int i=0; i<n; ++i){
sort(col[i].begin(), col[i].end(),
[&](int p, int q){return y[p] < y[q];});
}
vector<ll> up(m,0), down(m,0);
vector<ll> up_max(n,0), down_max(n,0);
for(int i=0; i<n; ++i){
if(col[i].size() == 0){
if(i==0){
up_max[i] = 0;
down_max[i] = 0;
}
else{
up_max[i] = up_max[i-1];
down_max[i] = down_max[i-1];
}
continue;
}
//case 1: delete (i-1)-st column and start over
ll mx = (i<2) ? 0LL : max(up_max[i-2], down_max[i-1]);
for(int j=0; j<col[i].size(); ++j){
if(j==0) up[col[i][0]] = mx + w[col[i][0]];
else{
up[col[i][j]] = up[col[i][j-1]] + w[col[i][j]];
}
}
if(i > 0){
mx = (i<2) ? 0LL : max(up_max[i-2], down_max[i-2]);
for(int j=col[i].size()-1; j>=0; --j){
if(j==col[i].size()-1) down[col[i][j]] = mx + w[col[i][j]];
else{
down[col[i][j]] = down[col[i][j+1]] + w[col[i][j]];
}
}
}
//case 2: actually using the previous column
if(i>0 && col[i-1].size() > 0){
int p, x = col[i-1].size();
p = -1;
for(int j=0; j<col[i].size(); j++){
while(p < x-1 && y[col[i-1][p+1]] < y[col[i][j]]) p++;
if(p==-1) continue;
if(j==0) up[col[i][j]] = max(up[col[i][j]], up[col[i-1][p]] + w[col[i][j]]);
else{
up[col[i][j]] = max({up[col[i][j]],
up[col[i-1][p]] + w[col[i][j]],
up[col[i][j-1]] + w[col[i][j]]});
}
}
p = x;
if(i>1){
for(int j=x-1; j>=0; j--){
while(p > 0 && y[col[i-1][p-1]] > y[col[i][j]]) p--;
if(p==x) continue;
if(j==x-1) down[col[i][j]] = max(down[col[i][j]], down[col[i-1][p]] + w[col[i][j]]);
else{
down[col[i][j]] = max({down[col[i][j]],
down[col[i-1][p]] + w[col[i][j]],
down[col[i][j+1]] + w[col[i][j]]});
}
}
}
}
if(i==0){
up_max[i] = up[col[i][col[i].size()-1]];
down_max[i] = down[col[i][0]];
}
else{
up_max[i] = max(up_max[i-1], up[col[i][col[i].size()-1]]);
down_max[i] = max(down_max[i-1], down[col[i][0]]);
}
}
/*for(int i=0; i<n; ++i){
cout << up_max[i] << " " << down_max[i] << "\n";
}*/
return max(up_max[n-2], down_max[n-1]);
}
Compilation message
fish.cpp: In function 'long long int max_weights(int, int, std::vector<int>, std::vector<int>, std::vector<int>)':
fish.cpp:31:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
31 | for(int j=0; j<col[i].size(); ++j){
| ~^~~~~~~~~~~~~~
fish.cpp:40:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
40 | if(j==col[i].size()-1) down[col[i][j]] = mx + w[col[i][j]];
| ~^~~~~~~~~~~~~~~~~
fish.cpp:52:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
52 | for(int j=0; j<col[i].size(); j++){
| ~^~~~~~~~~~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
31 ms |
7620 KB |
Output is correct |
2 |
Correct |
39 ms |
9016 KB |
Output is correct |
3 |
Correct |
3 ms |
4180 KB |
Output is correct |
4 |
Correct |
3 ms |
4180 KB |
Output is correct |
5 |
Correct |
120 ms |
18504 KB |
Output is correct |
6 |
Correct |
129 ms |
20336 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
66 ms |
11560 KB |
Output is correct |
3 |
Correct |
80 ms |
13752 KB |
Output is correct |
4 |
Correct |
35 ms |
7640 KB |
Output is correct |
5 |
Correct |
41 ms |
9024 KB |
Output is correct |
6 |
Correct |
0 ms |
212 KB |
Output is correct |
7 |
Correct |
0 ms |
212 KB |
Output is correct |
8 |
Correct |
0 ms |
256 KB |
Output is correct |
9 |
Correct |
0 ms |
212 KB |
Output is correct |
10 |
Correct |
3 ms |
4180 KB |
Output is correct |
11 |
Correct |
3 ms |
4180 KB |
Output is correct |
12 |
Correct |
33 ms |
7772 KB |
Output is correct |
13 |
Correct |
40 ms |
9024 KB |
Output is correct |
14 |
Correct |
32 ms |
7800 KB |
Output is correct |
15 |
Correct |
38 ms |
8668 KB |
Output is correct |
16 |
Correct |
35 ms |
7780 KB |
Output is correct |
17 |
Correct |
37 ms |
8612 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
3 ms |
4264 KB |
Output is correct |
2 |
Correct |
3 ms |
4180 KB |
Output is correct |
3 |
Correct |
24 ms |
7508 KB |
Output is correct |
4 |
Correct |
17 ms |
6760 KB |
Output is correct |
5 |
Correct |
46 ms |
11204 KB |
Output is correct |
6 |
Correct |
48 ms |
11264 KB |
Output is correct |
7 |
Correct |
46 ms |
11204 KB |
Output is correct |
8 |
Correct |
44 ms |
11212 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
0 ms |
212 KB |
Output is correct |
3 |
Correct |
0 ms |
212 KB |
Output is correct |
4 |
Correct |
0 ms |
212 KB |
Output is correct |
5 |
Correct |
0 ms |
212 KB |
Output is correct |
6 |
Correct |
0 ms |
212 KB |
Output is correct |
7 |
Correct |
0 ms |
212 KB |
Output is correct |
8 |
Correct |
0 ms |
212 KB |
Output is correct |
9 |
Correct |
1 ms |
212 KB |
Output is correct |
10 |
Correct |
1 ms |
340 KB |
Output is correct |
11 |
Runtime error |
1 ms |
468 KB |
Execution killed with signal 11 |
12 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
0 ms |
212 KB |
Output is correct |
3 |
Correct |
0 ms |
212 KB |
Output is correct |
4 |
Correct |
0 ms |
212 KB |
Output is correct |
5 |
Correct |
0 ms |
212 KB |
Output is correct |
6 |
Correct |
0 ms |
212 KB |
Output is correct |
7 |
Correct |
0 ms |
212 KB |
Output is correct |
8 |
Correct |
0 ms |
212 KB |
Output is correct |
9 |
Correct |
1 ms |
212 KB |
Output is correct |
10 |
Correct |
1 ms |
340 KB |
Output is correct |
11 |
Runtime error |
1 ms |
468 KB |
Execution killed with signal 11 |
12 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
0 ms |
212 KB |
Output is correct |
3 |
Correct |
0 ms |
212 KB |
Output is correct |
4 |
Correct |
0 ms |
212 KB |
Output is correct |
5 |
Correct |
0 ms |
212 KB |
Output is correct |
6 |
Correct |
0 ms |
212 KB |
Output is correct |
7 |
Correct |
0 ms |
212 KB |
Output is correct |
8 |
Correct |
0 ms |
212 KB |
Output is correct |
9 |
Correct |
1 ms |
212 KB |
Output is correct |
10 |
Correct |
1 ms |
340 KB |
Output is correct |
11 |
Runtime error |
1 ms |
468 KB |
Execution killed with signal 11 |
12 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
3 ms |
4264 KB |
Output is correct |
2 |
Correct |
3 ms |
4180 KB |
Output is correct |
3 |
Correct |
24 ms |
7508 KB |
Output is correct |
4 |
Correct |
17 ms |
6760 KB |
Output is correct |
5 |
Correct |
46 ms |
11204 KB |
Output is correct |
6 |
Correct |
48 ms |
11264 KB |
Output is correct |
7 |
Correct |
46 ms |
11204 KB |
Output is correct |
8 |
Correct |
44 ms |
11212 KB |
Output is correct |
9 |
Correct |
47 ms |
11212 KB |
Output is correct |
10 |
Correct |
40 ms |
7648 KB |
Output is correct |
11 |
Correct |
99 ms |
15952 KB |
Output is correct |
12 |
Correct |
0 ms |
212 KB |
Output is correct |
13 |
Correct |
0 ms |
212 KB |
Output is correct |
14 |
Correct |
0 ms |
212 KB |
Output is correct |
15 |
Correct |
0 ms |
212 KB |
Output is correct |
16 |
Correct |
0 ms |
212 KB |
Output is correct |
17 |
Correct |
0 ms |
212 KB |
Output is correct |
18 |
Correct |
3 ms |
4180 KB |
Output is correct |
19 |
Correct |
3 ms |
4180 KB |
Output is correct |
20 |
Correct |
3 ms |
4180 KB |
Output is correct |
21 |
Correct |
3 ms |
4180 KB |
Output is correct |
22 |
Incorrect |
49 ms |
10356 KB |
1st lines differ - on the 1st token, expected: '45561826463480', found: '45639306245662' |
23 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
31 ms |
7620 KB |
Output is correct |
2 |
Correct |
39 ms |
9016 KB |
Output is correct |
3 |
Correct |
3 ms |
4180 KB |
Output is correct |
4 |
Correct |
3 ms |
4180 KB |
Output is correct |
5 |
Correct |
120 ms |
18504 KB |
Output is correct |
6 |
Correct |
129 ms |
20336 KB |
Output is correct |
7 |
Correct |
0 ms |
212 KB |
Output is correct |
8 |
Correct |
66 ms |
11560 KB |
Output is correct |
9 |
Correct |
80 ms |
13752 KB |
Output is correct |
10 |
Correct |
35 ms |
7640 KB |
Output is correct |
11 |
Correct |
41 ms |
9024 KB |
Output is correct |
12 |
Correct |
0 ms |
212 KB |
Output is correct |
13 |
Correct |
0 ms |
212 KB |
Output is correct |
14 |
Correct |
0 ms |
256 KB |
Output is correct |
15 |
Correct |
0 ms |
212 KB |
Output is correct |
16 |
Correct |
3 ms |
4180 KB |
Output is correct |
17 |
Correct |
3 ms |
4180 KB |
Output is correct |
18 |
Correct |
33 ms |
7772 KB |
Output is correct |
19 |
Correct |
40 ms |
9024 KB |
Output is correct |
20 |
Correct |
32 ms |
7800 KB |
Output is correct |
21 |
Correct |
38 ms |
8668 KB |
Output is correct |
22 |
Correct |
35 ms |
7780 KB |
Output is correct |
23 |
Correct |
37 ms |
8612 KB |
Output is correct |
24 |
Correct |
3 ms |
4264 KB |
Output is correct |
25 |
Correct |
3 ms |
4180 KB |
Output is correct |
26 |
Correct |
24 ms |
7508 KB |
Output is correct |
27 |
Correct |
17 ms |
6760 KB |
Output is correct |
28 |
Correct |
46 ms |
11204 KB |
Output is correct |
29 |
Correct |
48 ms |
11264 KB |
Output is correct |
30 |
Correct |
46 ms |
11204 KB |
Output is correct |
31 |
Correct |
44 ms |
11212 KB |
Output is correct |
32 |
Correct |
0 ms |
212 KB |
Output is correct |
33 |
Correct |
0 ms |
212 KB |
Output is correct |
34 |
Correct |
0 ms |
212 KB |
Output is correct |
35 |
Correct |
0 ms |
212 KB |
Output is correct |
36 |
Correct |
0 ms |
212 KB |
Output is correct |
37 |
Correct |
0 ms |
212 KB |
Output is correct |
38 |
Correct |
0 ms |
212 KB |
Output is correct |
39 |
Correct |
0 ms |
212 KB |
Output is correct |
40 |
Correct |
1 ms |
212 KB |
Output is correct |
41 |
Correct |
1 ms |
340 KB |
Output is correct |
42 |
Runtime error |
1 ms |
468 KB |
Execution killed with signal 11 |
43 |
Halted |
0 ms |
0 KB |
- |