#include "fish.h"
#include <vector>
#include <algorithm>
using namespace std;
using ll = long long;
ll solve(vector<vector<pair<int, ll>>> &fish, int n)
{
ll res = 0;
vector<vector<ll>> dp0(fish.size()), dp1(fish.size());
for (int i = 0; i < fish.size(); ++i)
{
fish[i].emplace_back(n, 0);
dp0[i].resize(fish[i].size());
ll mx = 0;
if (i != 0)
{
for (int j = 0; j < fish[i - 1].size(); ++j)
{
mx = max(mx, dp1[i - 1][j]);
}
fill(dp0[i].begin(), dp0[i].end(), mx);
mx = 0;
}
for (int j = 0, k = 0; j < fish[i].size(); ++j)
{
if (i != 0)
{
while (k < fish[i - 1].size() && fish[i - 1][k].first < fish[i][j].first)
{
mx = max(mx, dp0[i - 1][k]);
mx += fish[i - 1][k].second;
++k;
}
}
dp0[i][j] = max(mx, dp0[i][j]);
}
dp1 = dp0;
if (i != 0)
{
ll mx = 0;
for (int j = (int)fish[i].size() - 1, k = (int)fish[i - 1].size() - 1; j >= 0; --j)
{
while (k >= 0 && fish[i - 1][k].first > fish[i][j].first)
{
mx = max(mx, dp1[i - 1][k]);
--k;
}
mx += fish[i][j].second;
dp1[i][j] = max(dp1[i][j], mx);
res = max(dp1[i][j], res);
}
}
}
return res;
}
ll max_weights(int n, int m, vector<int> X, vector<int> Y, vector<int> W)
{
vector<tuple<int, int, ll>> xyw(m);
for (int i = 0; i < m; ++i) xyw[i] = {X[i], Y[i], W[i]};
sort(xyw.begin(), xyw.end());
ll res = 0;
vector<vector<pair<int, ll>>> tos;
int prevX = -1;
for (auto [x, y, w] : xyw)
{
if (x > prevX + 1)
{
tos.push_back(vector<pair<int, ll>>());
res += solve(tos, n);
tos.clear();
tos.push_back(vector<pair<int, ll>>());
prevX = x - 1;
}
if (x == prevX + 1)
{
tos.push_back(vector<pair<int, ll>>());
}
tos.back().emplace_back(y, w);
prevX = x;
}
if (prevX != n - 1) tos.push_back(vector<pair<int, ll>>());
res += solve(tos, n);
return res;
}
Compilation message
fish.cpp: In function 'll solve(std::vector<std::vector<std::pair<int, long long int> > >&, int)':
fish.cpp:14:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::vector<std::pair<int, long long int> > >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
14 | for (int i = 0; i < fish.size(); ++i)
| ~~^~~~~~~~~~~~~
fish.cpp:21:31: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
21 | for (int j = 0; j < fish[i - 1].size(); ++j)
| ~~^~~~~~~~~~~~~~~~~~~~
fish.cpp:28:34: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
28 | for (int j = 0, k = 0; j < fish[i].size(); ++j)
| ~~^~~~~~~~~~~~~~~~
fish.cpp:32:26: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
32 | while (k < fish[i - 1].size() && fish[i - 1][k].first < fish[i][j].first)
| ~~^~~~~~~~~~~~~~~~~~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
31 ms |
5956 KB |
Output is correct |
2 |
Correct |
43 ms |
7324 KB |
Output is correct |
3 |
Correct |
0 ms |
212 KB |
Output is correct |
4 |
Correct |
0 ms |
212 KB |
Output is correct |
5 |
Correct |
126 ms |
15440 KB |
Output is correct |
6 |
Correct |
169 ms |
11988 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
212 KB |
Output is correct |
2 |
Correct |
68 ms |
11648 KB |
Output is correct |
3 |
Correct |
95 ms |
14400 KB |
Output is correct |
4 |
Correct |
32 ms |
5892 KB |
Output is correct |
5 |
Correct |
40 ms |
7268 KB |
Output is correct |
6 |
Correct |
0 ms |
212 KB |
Output is correct |
7 |
Correct |
1 ms |
212 KB |
Output is correct |
8 |
Correct |
1 ms |
212 KB |
Output is correct |
9 |
Correct |
0 ms |
212 KB |
Output is correct |
10 |
Correct |
0 ms |
212 KB |
Output is correct |
11 |
Correct |
0 ms |
212 KB |
Output is correct |
12 |
Correct |
39 ms |
5896 KB |
Output is correct |
13 |
Correct |
46 ms |
7268 KB |
Output is correct |
14 |
Correct |
34 ms |
5908 KB |
Output is correct |
15 |
Correct |
40 ms |
6592 KB |
Output is correct |
16 |
Correct |
34 ms |
5972 KB |
Output is correct |
17 |
Correct |
49 ms |
6596 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
0 ms |
212 KB |
Output is correct |
3 |
Correct |
127 ms |
2516 KB |
Output is correct |
4 |
Execution timed out |
1087 ms |
3500 KB |
Time limit exceeded |
5 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
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 |
1 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 |
1 ms |
212 KB |
Output is correct |
8 |
Incorrect |
1 ms |
212 KB |
1st lines differ - on the 1st token, expected: '2', found: '1' |
9 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
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 |
1 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 |
1 ms |
212 KB |
Output is correct |
8 |
Incorrect |
1 ms |
212 KB |
1st lines differ - on the 1st token, expected: '2', found: '1' |
9 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
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 |
1 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 |
1 ms |
212 KB |
Output is correct |
8 |
Incorrect |
1 ms |
212 KB |
1st lines differ - on the 1st token, expected: '2', found: '1' |
9 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
0 ms |
212 KB |
Output is correct |
3 |
Correct |
127 ms |
2516 KB |
Output is correct |
4 |
Execution timed out |
1087 ms |
3500 KB |
Time limit exceeded |
5 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
31 ms |
5956 KB |
Output is correct |
2 |
Correct |
43 ms |
7324 KB |
Output is correct |
3 |
Correct |
0 ms |
212 KB |
Output is correct |
4 |
Correct |
0 ms |
212 KB |
Output is correct |
5 |
Correct |
126 ms |
15440 KB |
Output is correct |
6 |
Correct |
169 ms |
11988 KB |
Output is correct |
7 |
Correct |
1 ms |
212 KB |
Output is correct |
8 |
Correct |
68 ms |
11648 KB |
Output is correct |
9 |
Correct |
95 ms |
14400 KB |
Output is correct |
10 |
Correct |
32 ms |
5892 KB |
Output is correct |
11 |
Correct |
40 ms |
7268 KB |
Output is correct |
12 |
Correct |
0 ms |
212 KB |
Output is correct |
13 |
Correct |
1 ms |
212 KB |
Output is correct |
14 |
Correct |
1 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 |
39 ms |
5896 KB |
Output is correct |
19 |
Correct |
46 ms |
7268 KB |
Output is correct |
20 |
Correct |
34 ms |
5908 KB |
Output is correct |
21 |
Correct |
40 ms |
6592 KB |
Output is correct |
22 |
Correct |
34 ms |
5972 KB |
Output is correct |
23 |
Correct |
49 ms |
6596 KB |
Output is correct |
24 |
Correct |
0 ms |
212 KB |
Output is correct |
25 |
Correct |
0 ms |
212 KB |
Output is correct |
26 |
Correct |
127 ms |
2516 KB |
Output is correct |
27 |
Execution timed out |
1087 ms |
3500 KB |
Time limit exceeded |
28 |
Halted |
0 ms |
0 KB |
- |