#include "fish.h"
#include <bits/stdc++.h>
using namespace std;
#define ff first
#define ss second
#define all(a) (a).begin(), (a).end()
#define ll long long
template<typename T>
int len(T &a){
return a.size();
}
mt19937 rng(chrono::high_resolution_clock::now().time_since_epoch().count());
long long max_weights(int n, int m, std::vector<int> x, std::vector<int> y,
std::vector<int> w) {
vector<vector<ll>> s(n + 1, vector<ll> (n + 1));
for(int i = 0; i < m; i ++) s[x[i] + 1][y[i] + 1] = w[i];
for(int i = 1; i <= n; i ++) for(int j = 1; j <= n; j ++) s[i][j] += s[i][j - 1];
vector<vector<array<ll, 2>>> dp(n + 1, vector<array<ll, 2>>(n + 1));
for(int j = 0; j <= n; j ++){
dp[2][j][0] = s[1][j];
}
for(int i = 2; i <= n; i ++){
ll mx = 0;
ll mx2 = 0;
vector<ll> suf(n);
for(int j = n; j >= 0; j --){
mx2 = max(mx2, max(dp[i - 2][j][0], dp[i - 2][j][1]) + s[i - 1][j]);
suf[j] = mx2;
}
mx2 = 0;
for(int j = 0; j <= n; j ++){
mx2 = max({mx2, dp[i - 2][j][0], dp[i - 2][j][1]});
if(i > 2) dp[i][j][0] = max(mx2 + s[i - 1][j], suf[j]);
dp[i][j][1] = dp[i][j][0];
mx = max(mx, dp[i - 1][j][0] - s[i - 1][j]);
dp[i][j][0] = max(dp[i][j][0], mx + s[i - 1][j]);
}
mx = 0;
for(int j = n; j >= 0; j --){
mx = max(mx, max(dp[i - 1][j][0], dp[i - 1][j][1]) - (s[i][n] - s[i][j]));
dp[i][j][1] = max(dp[i][j][1], mx + (s[i][n] - s[i][j]));
}
}
ll ans = 0;
for(int i = 0; i <= n; i ++){
ans = max({ans, dp[n][i][0], dp[n][i][1]});
//cout << dp[n][i][0] << ' ' << dp[n][i][1] << endl;
}
return ans;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
775 ms |
2097152 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
1 ms |
344 KB |
Execution killed with signal 6 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
806 ms |
2097152 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
348 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 |
1 ms |
348 KB |
Output is correct |
7 |
Runtime error |
0 ms |
348 KB |
Execution killed with signal 6 |
8 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
348 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 |
1 ms |
348 KB |
Output is correct |
7 |
Runtime error |
0 ms |
348 KB |
Execution killed with signal 6 |
8 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
348 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 |
1 ms |
348 KB |
Output is correct |
7 |
Runtime error |
0 ms |
348 KB |
Execution killed with signal 6 |
8 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
806 ms |
2097152 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
775 ms |
2097152 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |