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 <vector>
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define pii pair<int,int>
#define pll pair<ll,ll>
#define fs first
#define sc second
const ll inf = 1e18;
const int mxn = 330;
ll dp[mxn][mxn][2];
vector<pii> fish[mxn];
ll f(int idx,int s,int e){
ll re = 0;
for(auto &[pos,w]:fish[idx]){
if(pos>s&&pos<=e)re += w;
}
return re;
}
long long max_weights(int N, int M, std::vector<int> X, std::vector<int> Y,
std::vector<int> W) {
for(int i = 0;i<M;i++){
fish[X[i]].push_back(pii(Y[i],W[i]));
}
for(int i = 0;i<N;i++)sort(fish[i].begin(),fish[i].end());
for(int i = 0;i<N;i++){
dp[0][i][0] = f(1,0,i);
}
ll big = 0;
for(int i = 1;i<N;i++){
for(int h = 0;h<N;h++){
for(int j = 0;j<N;j++){
if(j<=h){
if(i-2>=0)dp[i][h][0] = max(dp[i][h][0],max(dp[i-2][j][0],dp[i-2][j][1])+f(i-1,j,h));
dp[i][h][1] = max(dp[i][h][1],dp[i-1][j][0]-f(i,-1,j)+f(i-1,j,h));
}
else{
if(i-2>=0)dp[i][h][0] = max(dp[i][h][0],max(dp[i-2][j][0],dp[i-2][j][1]));
dp[i][h][1] = max(dp[i][h][1],dp[i-1][j][0]-f(i,-1,j));
}
}
dp[i][h][0] = max(dp[i][h][0],big+f(i-1,-1,h));
dp[i][h][0] += f(i+1,-1,h);
dp[i][h][1] += f(i+1,-1,h);
}
if(i-2>=0){
for(int j = 0;j<N;j++)big = max({big,dp[i-2][j][0],dp[i-2][j][1]});
}
}
ll ans = 0;
/*
for(int i = 0;i<N;i++){
for(int j = 0;j<N;j++)cerr<<dp[i][j][0]<<','<<dp[i][j][1]<<' ';cerr<<endl;
}cerr<<endl;
*/
for(int i = 0;i<N;i++){
for(int j = 0;j<N;j++)ans = max({ans,dp[i][j][0],dp[i][j][1]});
}
return ans;
}
# | 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... |