#include "fish.h"
#include <vector>
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
constexpr ll INF = 1000000000000000000;
/*
to make life easier, the value of a pier is (wt of fish behind/in front) - (wt of fish on pier)
the path can be in a few states;
- back-facing (pier at x=i that catches some fish at i-1)
- front-facing (pier at x=i-1 that catches some fish at i)
a back-facing path at y=h can move to
- a back-facing path at some higher/same y value
- a front-facing path at the same y value (do NOT count the - value of this pier, and ADD it to cancel the - of the ff path)
a front-facing path at y=h can move to
- a front-facing path at some lower/(same) y value
- a back-facing path at some lower/same y value (do NOT count the + value of the first back-facing pier)
- a back-facing path at a higher/(same) y value (do NOT count the + value of this pier)
- stasis
stasis can move to any back-facing path, or to stasis
so ???? a dp here
start the x-coordinates at 1; starting state is stasis at x=0
pad 2 columns to the right; end state is stasis on last column
*/
long long max_weights(int N, int M, std::vector<int> X, std::vector<int> Y,
std::vector<int> W) {
vector<vector<ll>> weights(N + 3, vector<ll>(N));
for (int i = 0; i < M; ++i) {
weights[X[i] + 1][Y[i]] = W[i];
}
vector<vector<ll>> columnSum = weights;
for (int xc = 0; xc < columnSum.size(); ++xc) {
for (int yc = 1; yc < N; ++yc) {
columnSum[xc][yc] += columnSum[xc][yc-1];
}
}
// needs xc > 0
auto pierValue = [&] (int xc, int yc, int c1, int c2) {
return c1 * columnSum[xc-1][yc] + c2 * columnSum[xc][yc];
};
vector<ll> stasisOpt(N + 3, -INF);
vector<vector<ll>> backFaceOpt(N+3, vector<ll>(N, -INF)), frontFaceOpt(N+3, vector<ll>(N, -INF));
stasisOpt[N+2] = 0;
for (int xc = N+1; xc >= 0; --xc) {
for (int yc = 0; yc < N; ++yc) {
ll &b = backFaceOpt[xc][yc], &f = frontFaceOpt[xc][yc], &s = stasisOpt[xc];
if (0 < xc && xc <= N) {
// find value for the back-facing path here
b = max(b, frontFaceOpt[xc+1][yc] + pierValue(xc, yc, 1, 1));
for (int nyc = yc; nyc < N; ++nyc) {
b = max(b, backFaceOpt[xc+1][nyc] + pierValue(xc, yc, 1, -1));
}
}
if (2 <= xc && xc <= N + 1) {
// find value for the front-facing path here
f = max(f, stasisOpt[xc+1] + pierValue(xc, yc, -1, 1));
for (int nyc = 0; nyc <= yc; ++nyc) {
f = max(f, frontFaceOpt[xc+1][nyc] + pierValue(xc, yc, -1, 1));
f = max(f, backFaceOpt[xc+1][nyc] + pierValue(xc, yc, -1, 1) - pierValue(xc+1, nyc, 1, 0));
}
for (int nyc = yc; nyc < N; ++nyc) {
f = max(f, backFaceOpt[xc+1][nyc] + pierValue(xc, yc, -1, 0));
}
}
// find value for the stasis path here
s = max(s, stasisOpt[xc+1]);
for (int nyc = 0; nyc < N; ++nyc) {
s = max(s, backFaceOpt[xc+1][nyc]);
}
/*
printf("%d %d: back %lld front %lld stasis %lld ", xc, yc, b, f, s);
if (xc > 0) printf("(raw values) %lld %lld\n", columnSum[xc-1][yc], columnSum[xc][yc]);
*/
}
}
return stasisOpt[0];
}
Compilation message
fish.cpp: In function 'long long int max_weights(int, int, std::vector<int>, std::vector<int>, std::vector<int>)':
fish.cpp:41:25: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::vector<long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
41 | for (int xc = 0; xc < columnSum.size(); ++xc) {
| ~~~^~~~~~~~~~~~~~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
915 ms |
2097152 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
236 KB |
Output is correct |
2 |
Runtime error |
877 ms |
2097152 KB |
Execution killed with signal 9 |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
772 ms |
2097152 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
1 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 |
0 ms |
212 KB |
Output is correct |
8 |
Correct |
1 ms |
280 KB |
Output is correct |
9 |
Correct |
12 ms |
980 KB |
Output is correct |
10 |
Correct |
93 ms |
3256 KB |
Output is correct |
11 |
Correct |
14 ms |
980 KB |
Output is correct |
12 |
Correct |
108 ms |
3224 KB |
Output is correct |
13 |
Correct |
3 ms |
468 KB |
Output is correct |
14 |
Correct |
93 ms |
3208 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
1 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 |
0 ms |
212 KB |
Output is correct |
8 |
Correct |
1 ms |
280 KB |
Output is correct |
9 |
Correct |
12 ms |
980 KB |
Output is correct |
10 |
Correct |
93 ms |
3256 KB |
Output is correct |
11 |
Correct |
14 ms |
980 KB |
Output is correct |
12 |
Correct |
108 ms |
3224 KB |
Output is correct |
13 |
Correct |
3 ms |
468 KB |
Output is correct |
14 |
Correct |
93 ms |
3208 KB |
Output is correct |
15 |
Correct |
96 ms |
3156 KB |
Output is correct |
16 |
Correct |
3 ms |
468 KB |
Output is correct |
17 |
Correct |
106 ms |
5032 KB |
Output is correct |
18 |
Correct |
105 ms |
5016 KB |
Output is correct |
19 |
Correct |
103 ms |
4948 KB |
Output is correct |
20 |
Correct |
108 ms |
5004 KB |
Output is correct |
21 |
Correct |
121 ms |
4984 KB |
Output is correct |
22 |
Correct |
113 ms |
6704 KB |
Output is correct |
23 |
Correct |
93 ms |
3540 KB |
Output is correct |
24 |
Correct |
99 ms |
4280 KB |
Output is correct |
25 |
Correct |
108 ms |
3236 KB |
Output is correct |
26 |
Correct |
97 ms |
3480 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
1 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 |
0 ms |
212 KB |
Output is correct |
8 |
Correct |
1 ms |
280 KB |
Output is correct |
9 |
Correct |
12 ms |
980 KB |
Output is correct |
10 |
Correct |
93 ms |
3256 KB |
Output is correct |
11 |
Correct |
14 ms |
980 KB |
Output is correct |
12 |
Correct |
108 ms |
3224 KB |
Output is correct |
13 |
Correct |
3 ms |
468 KB |
Output is correct |
14 |
Correct |
93 ms |
3208 KB |
Output is correct |
15 |
Correct |
96 ms |
3156 KB |
Output is correct |
16 |
Correct |
3 ms |
468 KB |
Output is correct |
17 |
Correct |
106 ms |
5032 KB |
Output is correct |
18 |
Correct |
105 ms |
5016 KB |
Output is correct |
19 |
Correct |
103 ms |
4948 KB |
Output is correct |
20 |
Correct |
108 ms |
5004 KB |
Output is correct |
21 |
Correct |
121 ms |
4984 KB |
Output is correct |
22 |
Correct |
113 ms |
6704 KB |
Output is correct |
23 |
Correct |
93 ms |
3540 KB |
Output is correct |
24 |
Correct |
99 ms |
4280 KB |
Output is correct |
25 |
Correct |
108 ms |
3236 KB |
Output is correct |
26 |
Correct |
97 ms |
3480 KB |
Output is correct |
27 |
Execution timed out |
1109 ms |
282920 KB |
Time limit exceeded |
28 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
772 ms |
2097152 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
915 ms |
2097152 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |