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 <bits/stdc++.h>
#define int long long
using namespace std;
map<pair<int,int>, int> a;
vector<vector<int>> fsh;
map<pair<int,pair<int,int>>, int> memo;
int N,M;
int dp(int x, int last, int big){
if(memo.find({x,{last,big}})!=memo.end()) return memo[{x,{last,big}}];
if(x==N) return 0;
memo[{x,{last,big}}] = 0;
for (auto i:fsh[x])
{
int asc=big;
int c=0;
if(i>last) {
if(big==0) break;
c=dp(x+1, i, 1);
asc=true;
}else if(i<last){
asc=false;
c=dp(x+1, i, 0);
}else if(i==last) {
c=dp(x+1, i, big);
}
if(x>0) {
if(asc) for (int y = last; y < i; y++) c+=a[{x-1,y}];
else for (int y = i; y < last; y++) c+=a[{x,y}];
}
memo[{x,{last,big}}]=max(c, memo[{x,{last,big}}]);
}
return memo[{x,{last,big}}];
}
long long max_weights(signed n, signed m, std::vector<signed> X, std::vector<signed> Y, std::vector<signed> W) {
N=n; M=m;
fsh.resize(N+1, vector<int>(1,0));
for (int i = 0; i < M; i++){
a[{X[i],Y[i]}] = W[i];
fsh[X[i]+1].push_back(Y[i]+1);
if(X[i]>0) fsh[X[i]-1].push_back(Y[i]+1);
}
for (int i = 0; i < N; i++) sort(fsh[i].begin(),fsh[i].end());
int p=dp(0,0,1);
return p;
}
# | 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... |