#include "fish.h"
#include<bits/stdc++.h>
using namespace std;
#define int long long int
int pref[3005][3005];
int query(int j, int i1, int i2) {
if (i1 > i2) {
return 0;
}
return pref[i2][j]-pref[i1-1][j];
}
int max_weights(int32_t N, int32_t M, vector<int32_t> X, vector<int32_t> Y, vector<int32_t> W) {
int n = N, m = M, yn;
yn = N;
vector<vector<int>> c(yn+1, vector<int>(n+1, 0));
vector<vector<vector<int>>> dp(yn+1, vector<vector<int>>(n+1, vector<int>(2, 0))); // 0 is increasing, 1 is decreasing, increasing can go to decreasing, decreasing cant go to increasing if next to eachother
for (int i = 0; i < m; i++) {
c[Y[i]+1][X[i]] = W[i];
pref[Y[i]+1][X[i]] = W[i];
}
for (int j = 0; j < n; j++) {
for (int i = 1; j <= yn; j++) {
pref[i][j] += pref[i-1][j];
}
}
for (int j = 1; j < n; j++) {
int v = 0;
dp[0][j][0] = max(dp[0][j][0], dp[0][j-1][0]);
for (int i = 1; i <= yn; i++) {
v += c[i][j];
dp[0][j][0] = max(dp[0][j][0], dp[i][j-1][0]+v);
dp[0][j][0] = max(dp[0][j][0], dp[i][j-1][1]+v);
}
for (int i = 1; i <= yn; i++) {
for (int a = i; a <= yn; a++) {
v = query(j, i+1, a);
dp[i][j][1] = max(dp[i][j][1], dp[a][j-1][0]+v);
dp[i][j][1] = max(dp[i][j][1], dp[a][j-1][1]+v);
}
for (int a = 1; a <= i; a++) {
v = query(j-1, a+1, i);
dp[i][j][0] = max(dp[i][j][0], dp[a][j-1][0]+v);
}
}
if (j == 1) {
v = 0;
for (int i = 1; i <= yn; i++) {
v += c[i][j-1];
dp[i][j][0] = max(dp[i][j][0], dp[0][j-1][0]+v);
dp[i][j][1] = max(dp[i][j][1], dp[0][j-1][0]+v);
}
}
if (j-2 >= 0) {
v = 0;
for (int i = 1; i <= yn; i++) {
v += c[i][j-1];
dp[i][j][0] = max(dp[i][j][0], dp[0][j-2][0]+v);
dp[i][j][1] = max(dp[i][j][1], dp[0][j-2][0]+v);
}
for (int i = 1; i <= yn; i++) {
v = 0;
for (int h = 1; h <= i; h++) {
v += c[h][j-1];
}
for (int h = 1; h <= i; h++) {
dp[i][j][0] = max(dp[i][j][0], dp[h][j-2][0]+v);
dp[i][j][0] = max(dp[i][j][0], dp[h][j-2][1]+v);
dp[i][j][1] = max(dp[i][j][1], dp[h][j-2][0]+v);
dp[i][j][1] = max(dp[i][j][1], dp[h][j-2][1]+v);
}
for (int h = i+1; h < yn; h++) {
v += c[h][j-1];
dp[i][j][0] = max(dp[i][j][0], dp[h][j-2][0]+v);
dp[i][j][0] = max(dp[i][j][0], dp[h][j-2][1]+v);
dp[i][j][1] = max(dp[i][j][1], dp[h][j-2][0]+v);
dp[i][j][1] = max(dp[i][j][1], dp[h][j-2][1]+v);
}
}
}
}
int ans = 0;
for (int i = 0; i <= yn; i++) {
ans = max(ans, dp[i][n-1][0]);
ans = max(ans, dp[i][n-1][1]);
}
return ans;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
866 ms |
2097152 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
348 KB |
Output is correct |
2 |
Runtime error |
831 ms |
2097152 KB |
Execution killed with signal 9 |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
812 ms |
2097152 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
344 KB |
Output is correct |
2 |
Correct |
0 ms |
348 KB |
Output is correct |
3 |
Correct |
0 ms |
348 KB |
Output is correct |
4 |
Correct |
0 ms |
348 KB |
Output is correct |
5 |
Correct |
0 ms |
348 KB |
Output is correct |
6 |
Correct |
0 ms |
344 KB |
Output is correct |
7 |
Correct |
0 ms |
344 KB |
Output is correct |
8 |
Correct |
1 ms |
344 KB |
Output is correct |
9 |
Correct |
24 ms |
1884 KB |
Output is correct |
10 |
Incorrect |
209 ms |
6236 KB |
1st lines differ - on the 1st token, expected: '799839985182', found: '798600481950' |
11 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
344 KB |
Output is correct |
2 |
Correct |
0 ms |
348 KB |
Output is correct |
3 |
Correct |
0 ms |
348 KB |
Output is correct |
4 |
Correct |
0 ms |
348 KB |
Output is correct |
5 |
Correct |
0 ms |
348 KB |
Output is correct |
6 |
Correct |
0 ms |
344 KB |
Output is correct |
7 |
Correct |
0 ms |
344 KB |
Output is correct |
8 |
Correct |
1 ms |
344 KB |
Output is correct |
9 |
Correct |
24 ms |
1884 KB |
Output is correct |
10 |
Incorrect |
209 ms |
6236 KB |
1st lines differ - on the 1st token, expected: '799839985182', found: '798600481950' |
11 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
344 KB |
Output is correct |
2 |
Correct |
0 ms |
348 KB |
Output is correct |
3 |
Correct |
0 ms |
348 KB |
Output is correct |
4 |
Correct |
0 ms |
348 KB |
Output is correct |
5 |
Correct |
0 ms |
348 KB |
Output is correct |
6 |
Correct |
0 ms |
344 KB |
Output is correct |
7 |
Correct |
0 ms |
344 KB |
Output is correct |
8 |
Correct |
1 ms |
344 KB |
Output is correct |
9 |
Correct |
24 ms |
1884 KB |
Output is correct |
10 |
Incorrect |
209 ms |
6236 KB |
1st lines differ - on the 1st token, expected: '799839985182', found: '798600481950' |
11 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
812 ms |
2097152 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
866 ms |
2097152 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |