#include "fish.h"
#include<bits/stdc++.h>
using namespace std;
long long max_weights(int n, int m, vector<int> x, vector<int> y, vector<int> w){
vector<vector<pair<long long, long long> > > p;
vector<vector<vector<long long> > > dp;
vector<vector<long long> > pre;
vector<long long> ans;
ans.resize(n+10);
pre.resize(n+10);
dp.resize(n+10);
p.resize(n+10);
for(long long i=0; i<=n; i++) p[i].push_back(make_pair(0, 0));
for(long long i=0; i<m; i++) p[x[i]].push_back(make_pair(y[i], w[i]));
for(long long i=1; i<=n; i++){
sort(p[i].begin(), p[i].end());
dp[i].resize(p[i].size(), vector<long long>(5));
pre[i].resize(p[i].size());
for(long long j=1; j<p[i].size(); j++) pre[i][j]+=pre[i][j-1]+p[i][j].second;
}
for(long long i=2; i<=n; i++){
dp[i][0][0]=dp[i-1][0][0];
long long L=p[i-1].size();
long long val=dp[i-1][L][0]+pre[i].back();
for(long long j=pre[i].size()-1; j>=0; j--){
while(L && p[i-1][L-1].first>p[i][j].first) L--, val=max(val, dp[i-1][L][0]+pre[i][j]);
if(j) dp[i][j][3]=max(dp[i][j][3], val-pre[i][j-1]);
else dp[i][j][3]=max(dp[i][j][3], val);
}
dp[i][0][0]=max(dp[i][0][0], dp[i][0][3]);
long long D=0; val=-1e18;
for(long long j=1; j<p[i].size(); j++){
dp[i][j][1]=dp[i-1][0][0];
while(D+1<p[i-1].size() && p[i-1][D+1].first<p[i][j].first){
D++; val=max(val, dp[i-1][D][2]-pre[i-1][D-1]);
}
dp[i][j][2]=max(dp[i][j][2], max(val, ans[i-2])+pre[i-1][D]);
}
ans[i]=dp[i][0][0];
for(long long j=1; j<p[i].size(); j++){
dp[i][j][0]=max(dp[i][j][1], max(dp[i][j][2], dp[i][j][3]));
ans[i]=max(ans[i], dp[i][j][0]);
}
}
return ans[n];
}
//signed main(){
// return 0;
//}
# | 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... |