#include <bits/stdc++.h>
#include "fish.h"
using namespace std;
int n,m;
long long max_weights(int N, int M, vector<int> X,vector<int>Y,vector<int>W)
{
n=N;m=M;
vector<long long>pref1(n+1),suf1(n+1),suf2(n+1),pref2(n+1);
vector<vector<long long>>pref(n+1,vector<long long>(n+1));
vector<vector<vector<long long>>>dp(n,vector<vector<long long>>(n+1,vector<long long>(2)));
for(int i=0;i<m;i++)
{
pref[X[i]][Y[i]+1]+=W[i];
}
for(int i=0;i<n;i++)
{
for(int j=1;j<=n;j++)
{
pref[i][j]+=pref[i][j-1];
}
}
for(int i=1;i<n;i++)
{
for(int j=0;j<=n;j++)
{
dp[i][j][0]=max(dp[i][j][0],pref1[max(0,j-1)]+pref[i-1][j]);
dp[i][j][1]=max(dp[i][j][1],suf1[j]-pref[i][j]);
dp[i][j][0]=max(dp[i][j][0],max(suf2[j],pref2[j]+pref[i-1][j]));
dp[i][j][1]=max(dp[i][j][1],max(suf2[j],pref2[j]+pref[i-1][j]));
}
if(i==n-1)continue;
pref1[0]=dp[i][0][0]-pref[i][0];
pref2[0]=max(dp[i-1][0][0],dp[i-1][0][1]);
for(int j=1;j<=n;j++)
{
pref1[j]=max(pref1[j-1],dp[i][j][0]-pref[i][j]);
pref2[j]=max(pref2[j-1],max(dp[i-1][j][0],dp[i-1][j][1]));
}
suf1[n]=max(dp[i][n][0],dp[i][n][1])+pref[i+1][n];
suf2[n]=max(dp[i-1][n][0],dp[i-1][n][1])+pref[i][n];
for(int j=n-1;j>=0;j--)
{
suf1[j]=max(suf1[j+1],max(dp[i][j][0],dp[i][j][1])+pref[i+1][j]);
suf2[j]=max(suf2[j+1],max(dp[i-1][j][0],dp[i-1][j][1])+pref[i][j]);
}
}
long long res=0;
for(int i=0;i<=n;i++)
{
res=max(res,dp[n-1][i][0]);
res=max(res,dp[n-1][i][1]);
}
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... |