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;
#define OPTM ios_base::sync_with_stdio(0); cin.tie(0);
#define INF int(1e9+7)
#define ln '\n'
#define ll long long
#define ull unsigned long long
#define ui unsigned int
#define us unsigned short
#define FOR(i,s,n) for (int i = s; i < n; i++)
#define FORR(i,n,s) for (int i = n; i > s; i--)
#define FORX(u, arr) for (auto u : arr)
#define PB push_back
#define in(v,x) (v.find(x) != v.end())
#define F first
#define S second
#define PII pair<int, int>
#define PLL pair<ll, ll>
#define UM unordered_map
#define US unordered_set
#define PQ priority_queue
#define ALL(v) v.begin(), v.end()
const ll LLINF = 1e18+1;
#define int long long
const int MAXN = 3e2+5, MAXX = 15;
int n,m;
int dp[MAXN][MAXN], g[MAXN][MAXN], p[MAXN][MAXN];
int dp1[MAXN][MAXN], dp2[MAXN][MAXN], dp3[MAXN][MAXN], dp4[MAXN][MAXN], dp5[MAXN][MAXN];
long long max_weights(int32_t N, int32_t M, std::vector<int32_t> X, std::vector<int32_t> Y, std::vector<int32_t> W) {
n = N; m = M;
int yn = 0LL;
FOR(i,0,m) {
g[X[i]][Y[i]] = W[i]*1LL;
yn = max(yn,Y[i]*1LL);
}
FOR(i,1,n+1) {
FOR(j,1,n+1) {
p[i][j] = p[i][j-1]+g[i-1][j-1];
}
}
// FORR(j,yn+1,0) {
// FOR(i,1,n+1) cout << p[i][j] << " ";
// cout << ln;
// }
// cout << ln;
FOR(i,0,n+1) {
FOR(j,0,n+1) {
dp2[i][j] = -LLINF;
}
}
FOR(i,2,n+1) {
dp[i][0] = dp3[i-1][0];
FOR(j,1,n+1) {
dp[i][j] = dp2[i-1][j]+p[i-1][j];
dp1[i][j] = dp3[i-1][j]-p[i][j];
dp[i][j] = max(dp[i][j], max(dp3[i-2][j], dp4[i-2][j]+p[i-1][j])); // prev = 0 height
dp1[i][j] = max(dp1[i][j], max(dp3[i-2][j], dp4[i-2][j]+p[i-1][j])); // prev = 0 height
}
dp4[i][0] = max(dp[i][0], dp1[i][0]);
FOR(j,1,n+1) {
dp2[i][j] = max(dp2[i][j-1], dp[i][j]-p[i][j]);
dp4[i][j] = max(dp4[i][j-1], max(dp[i][j], dp1[i][j]));
}
dp3[i][n] = dp[i][n]+p[i+1][n];
FORR(j,n-1,-1) {
dp3[i][j] = max(dp3[i][j+1], max(dp[i][j], dp1[i][j])+p[i+1][j]);
}
// FOR(k,0,yn+2) {
// if (j >= k) {
// dp[i][j][k] = dp1[i-1][k]+p[i-1][j]-p[i-1][k];
// //dp[i][j][k] = max(dp[i][j][k], dp2[i-1][k][j]+p[i-1][j]);
// dp[i][j][k] = max(dp[i][j][k], dp3[i-1][k][j]);
// }
// else {
// dp[i][j][k] = dp3[i-1][k][0]+p[i][k]-p[i][j];
// }
// // FOR(l,0,yn+2) {
// // if (j >= k) {
// // dp[i][j][k] = max(dp[i][j][k], dp[i-1][k][l]+max(0LL,p[i-1][j]-p[i-1][max(k,l)]));
// // }
// // else {
// // dp[i][j][k] = max(dp[i][j][k], dp[i-1][k][l]+p[i][k]-p[i][j]);
// // }
// // }
// }
// FOR(j,0,yn+2) {
// FOR(k,0,yn+2) {
// if (k <= j) dp1[i][j] = max(dp1[i][j], dp[i][j][k]);
// else dp2[i][j][k] = max(dp2[i][j][k-1], dp[i][j][k]-p[i][k]);
// }
// FORR(k,yn+1,-1) {
// dp3[i][j][k] = max(dp3[i][j][k+1], dp[i][j][k]);
// }
// }
}
// cout << ln;
// FORR(j,yn+1,-1) {
// FOR(i,1,n+1) cout << max(dp[i][j], dp1[i][j]) << " ";
// cout << ln;
// }
int res = -LLINF;
FOR(j,0,n+1) {
res = max(res, max(dp[n][j], dp1[n][j]));
}
return res;
}
# | 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... |