#include "fish.h"
#include "bits/stdc++.h";
using namespace std;
#define forR(i, x) for(int i = 0; i < (x); ++i)
#define REP(i, a, b) for(int i = (a); i < (b); ++i)
typedef long long ll;
typedef vector<ll> vll;
typedef vector<vll> vvll;
typedef vector<int> vi;
typedef vector<vi> vvi;
struct pii{int a, b;};
ll INF = 1e16;
long long max_weights(int N, int M, std::vector<int> X, std::vector<int> Y,
std::vector<int> W) {
int MH = Y[0] + 1;
for(int i : Y) MH = max(MH, i + 1);
vvll wp(N, vll(MH));
vvll wpsa(N, vll(MH));
forR(x, N) {
wpsa[x][0] = wp[x][0];
REP(y, MH) wpsa[x][y] = wpsa[x][y-1] + wp[x][y];
}
forR(i, M) wp[X[i]][Y[i]] = W[i];
vvll I(N, vll(MH+1)), D(N, vll(MH+1)), E(N, vll(MH+1));
// h = HEIGHT, not the index of last element!
forR(i, N) {
REP(h, 1, MH+1) {
if(i == 0) I[i][h] = 0;
else {
forR(k, MH+1) {
ll tot = E[i-1][k] + wpsa[i-1][h-1] - (k > 0 ? wpsa[i-1][k-1] : 0);
// REP(y, k, h) tot += wp[i-1][y];
I[i][h] = max(I[i][h], tot);
}
REP(k, 1, h+1) {
ll tot = I[i-1][k] + wpsa[i-1][h-1] - (k > 0 ? wpsa[i-1][k-1] : 0);
// REP(y, k, h) tot += wp[i-1][y];
I[i][h] = max(I[i][h], tot);
}
}
}
REP(h, 1, MH+1) {
if(i == 0) D[i][h] = 0;
else {
REP(k, h+1, MH+1) {
ll tot = max(I[i-1][k], D[i-1][k]) + wpsa[i][k-1] - (h > 0 ? wpsa[i][h-1] : 0);
// REP(y, h, k) tot += wp[i][y];
D[i][h] = max(D[i][h], tot);
}
}
}
forR(h, MH+1) {
if(i == 0) E[i][h] = 0;
else {
ll tot = max(I[i-1][h], D[i-1][h]) + wpsa[i][h-1];
// forR(y, h) tot += wp[i][y];
E[i][h] = max(E[i][h], tot);
forR(y, MH+1) E[i][h] = max(E[i][h], E[i-1][y]);
}
}
}
ll mx = 0;
REP(h, 1, MH+1) mx = max(mx, max(I[N-1][h], D[N-1][h]));
forR(h, MH+1) mx = max(mx, E[N-1][h]);
return mx;
}
Compilation message
fish.cpp:2:25: warning: extra tokens at end of #include directive
2 | #include "bits/stdc++.h";
| ^
fish.cpp:23:14: error: macro "REP" requires 3 arguments, but only 2 given
23 | REP(y, MH) wpsa[x][y] = wpsa[x][y-1] + wp[x][y];
| ^
fish.cpp:5: note: macro "REP" defined here
5 | #define REP(i, a, b) for(int i = (a); i < (b); ++i)
|
fish.cpp: In function 'long long int max_weights(int, int, std::vector<int>, std::vector<int>, std::vector<int>)':
fish.cpp:23:5: error: 'REP' was not declared in this scope
23 | REP(y, MH) wpsa[x][y] = wpsa[x][y-1] + wp[x][y];
| ^~~
