#include "fish.h"
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef vector<int> vi;
typedef pair<int, int> ii;
typedef vector<ii> vii;
typedef long double ld;
typedef pair<ll, ll> pll;
#define FOR(i, a, b) for(int i = a; i < b; i++)
#define ROF(i, a, b) for(int i = a; i >= b; i--)
#define ms memset
#define pb push_back
#define fi first
#define se second
// O(n^2) dp
ll max_weights(int n, int m, vi X, vi Y, vi W){
ll mat[n][n]; //position of all fishes
ms(mat, 0, sizeof(mat));
FOR(i,0,m){
mat[X[i]][Y[i]] = W[i];
}
ll dp[n][n][2];
// dp[i][j][0] - mat[i][j] chosen, while increasing
// dp[i][j][1] - while decreasing
ms(dp,0,sizeof(dp));
FOR(i,0,n){
//decreasing
if (i > 0){
FOR(k,i-3,i-1){
dp[i][n-1][1] = max(dp[i][n-1][1], dp[k][n-1][0]);
dp[i][n-1][1] = max(dp[i][n-1][1], dp[k][0][1]);
}
dp[i][n-1][1] += mat[i][n-1];
ROF(j,n-2,0){
dp[i][j][1] = max(dp[i][j][1], dp[i][j+1][1]);
dp[i][j][1] = max(dp[i][j][1], dp[i-1][j+1][1]);
dp[i][j][1] += mat[i][j];
}
}
if (i < n-1){
//increasing
if (i>1) dp[i][0][0] = max(dp[i][0][0], dp[i-1][0][1]);
if (i>2) dp[i][0][0] = max(dp[i][0][0], dp[i-2][n-1][0]);
dp[i][0][0] += mat[i][0];
FOR(j,1,n){
dp[i][j][0] = max(dp[i][j][0], dp[i][j-1][0]);
if (i>0){
dp[i][j][0] = max(dp[i][j][0], dp[i-1][j-1][0]);
}
dp[i][j][0] += mat[i][j];
}
}
}
ll ans = 0;
FOR(i,0,n){
FOR(j,0,n){
ans = max(ans, max(dp[i][j][0], dp[i][j][1]));
}
}
return ans;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
751 ms |
2097152 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
212 KB |
Output is correct |
2 |
Runtime error |
785 ms |
2097152 KB |
Execution killed with signal 9 |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
738 ms |
2097152 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Incorrect |
0 ms |
212 KB |
1st lines differ - on the 1st token, expected: '7', found: '70' |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Incorrect |
0 ms |
212 KB |
1st lines differ - on the 1st token, expected: '7', found: '70' |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Incorrect |
0 ms |
212 KB |
1st lines differ - on the 1st token, expected: '7', found: '70' |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
738 ms |
2097152 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
751 ms |
2097152 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |