#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<vector<long long>>pref(n,vector<long long>(n+1));
vector<vector<vector<long long>>>dp(n,vector<vector<long long>>(n+1,vector<long long>(2))),dppref(n+1,vector<vector<long long>>(2,vector<long long>(2))),dpsuf(n+1,vector<vector<long long>>(2,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],dppref[j][0][0]+pref[i-1][j]);
dp[i][j][1]=max(dp[i][j][1],max(dpsuf[j][0][0],dpsuf[j][1][0])-pref[i][j]);
if(i>1)
{
dp[i][j][0]=max(dp[i][j][0],max(dppref[j][0][1],dppref[j][1][1])+pref[i-1][j]);
dp[i][j][0]=max(dp[i][j][0],max(dpsuf[j][0][1],dpsuf[j][1][1]));
dp[i][j][1]=max(dp[i][j][1],max(dppref[j][0][1],dppref[j][1][1])+pref[i-1][j]);
dp[i][j][1]=max(dp[i][j][1],max(dpsuf[j][0][1],dpsuf[j][1][1]));
}
}
if(i==n-1)continue;
dppref[0][0][0]=dp[i][0][0]-pref[i][0];
dpsuf[n][0][0]=dp[i][n][0]+pref[i+1][n];
dppref[0][1][0]=dp[i][0][1]-pref[i][0];
dpsuf[n][1][0]=dp[i][n][1]+pref[i+1][n];
dppref[0][0][1]=dp[i-1][0][0];
dpsuf[n][0][1]=dp[i-1][n][0]+pref[i][n];
dppref[0][1][1]=dp[i-1][0][1];
dpsuf[n][1][1]=dp[i-1][n][1]+pref[i][n];
for(int j=1;j<=n;j++)
{
dppref[j][0][0]=max(dp[i][j][0]-pref[i][j],dppref[j-1][0][0]);
dppref[j][1][0]=max(dp[i][j][1]-pref[i][j],dppref[j-1][1][0]);
dppref[j][0][1]=max(dppref[j-1][0][1],dp[i-1][j][0]);
dppref[j][1][1]=max(dppref[j-1][1][1],dp[i-1][j][1]);
}
for(int j=n-1;j>=0;j--)
{
dpsuf[j][0][0]=max(dp[i][j][0]+pref[i+1][j],dpsuf[j+1][0][0]);
dpsuf[j][1][0]=max(dp[i][j][1]+pref[i+1][j],dpsuf[j+1][1][0]);
dpsuf[j][0][1]=max(dp[i-1][j][0]+pref[i][j],dpsuf[j+1][0][1]);
dpsuf[j][1][1]=max(dp[i-1][j][1]+pref[i][j],dpsuf[j+1][1][1]);
}
}
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... |