This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include "fish.h"
#include <bits/stdc++.h>
using namespace std;
#pragma GCC optimize("Ofast")
#define loop(x, i) for (int i = 0; i < (x); i++)
#define loop1(x, i) for (int i = 1; i <= (x); i++)
#define rev(x, i) for (int i = (int)(x) - 1; i >= 0; i--)
#define itloop(x) for (auto it = begin(x); x != end(x); it++)
#define itrev(x) for (auto it = rbegin(x); x != rend(x); it++)
#define int long long
#define INF ((int64_t)(4e18 + 1))
#define INF32 ((int32_t)(2e9 + 1))
#define ALL(x) begin(x), end(x)
#define RALL(x) rbegin(x), rend(x)
#define removeIn(x, l) l.erase(find(ALL(l), x))
#define pb push_back
#define sz(x) (int)(x).size()
#define F first
#define S second
#define var const auto &
#define foreach(l) for (var e : l)
typedef int8_t i8;
typedef int16_t i16;
typedef int32_t i32;
typedef int64_t i64;
typedef pair<int, int> ii;
typedef tuple<int, int, int> iii;
typedef tuple<int, int, int, int> iiii;
typedef vector<int> vi;
typedef vector<i32> vi32;
typedef vector<vi> vvi;
typedef vector<vvi> vvvi;
typedef vector<vvvi> vvvvi;
typedef vector<ii> vii;
typedef vector<iii> viii;
typedef vector<vii> vvii;
typedef vector<viii> vviii;
typedef set<int> si;
typedef set<ii> sii;
typedef set<iii> siii;
typedef vector<si> vsi;
typedef vector<sii> vsii;
typedef vector<vsi> vvsi;
typedef vector<string> vstr;
typedef vector<vector<string>> vvstr;
typedef vector<bool> vb;
typedef vector<vb> vvb;
int max_weights(i32 N, i32 M, vi32 X, vi32 Y, vi32 W)
{
if (M == 0)
return 0;
bool allEven = 1;
for (int x : X)
if (x % 2)
allEven = 0;
if (allEven)
return accumulate(ALL(W), 0LL);
N = min(*max_element(ALL(X)) + 2, N);
int yCount = *max_element(ALL(Y)) + 2;
vvi wSum(N, vi(yCount));
{
vvi wsAtX(N, vi(yCount));
loop(M, i)
{
wsAtX[X[i]][Y[i]] = W[i];
}
loop(N, x)
{
loop1(yCount - 1, y)
{
wSum[x][y] = wSum[x][y - 1] + wsAtX[x][y - 1];
}
}
}
vvvi dp(vvvi(yCount, vvi(yCount, vi(2)))); // 0: adjacentL, 1: adjacentR
vvvi dp2(vvvi(yCount, vvi(yCount, vi(2)))); // 0: adjacentL, 1: adjacentR
vi maxAll(N);
vvi maxOfHeight(N, vi(yCount));
loop(N, x)
{
vvi pMax1perJ(yCount, vi(yCount)); // dp[<=j][i][1]
if (x != 0)
{
loop(yCount, j)
{
loop(yCount, i)
{
pMax1perJ[j][i] = max(j == 0 ? 0 : pMax1perJ[j - 1][i], dp[j][i][1]);
}
}
}
loop(yCount, j)
{
int prevMax = 0;
if (x != 0)
{
loop(yCount, i)
{
prevMax = max(prevMax, dp[j][i][0]);
}
}
loop(yCount, i) // hoogte van deze steiger
{
if (j <= i)
{
dp2[i][j][0] = dp2[i][j][1] = (x == 0) ? 0 : maxOfHeight[x - 1][i];
}
else
{
int sum = wSum[x][j] - wSum[x][i];
if (x == 0)
{
dp2[i][j][0] = -1;
dp2[i][j][1] = sum;
}
else
{
dp2[i][j][0] = max(dp[j][i][1], prevMax) + sum;
dp2[i][j][1] = max(pMax1perJ[j][i], maxAll[x - 1]) + sum;
}
}
maxAll[x] = max({maxAll[x], dp2[i][j][0]});
maxOfHeight[x][j] = max({maxOfHeight[x][j], dp2[i][j][0], dp2[i][j][1]});
}
}
dp = dp2;
}
return maxAll.back();
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |