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 ll long long
#include <vector>
ll mx(ll a,ll b){
if(a>b) return a;
return b;
}
long long max_weights(int N, int M, std::vector<int> X, std::vector<int> Y,std::vector<int> W) {
ll dp[N][2],a[N];
ll sum=0,say=0,sum0=0,sum1=0,say1=0;
for(int i=0;i<N;i++){
dp[i][0]=dp[i][1]=a[i]=0;
}
for(int i=0;i<M;i++){
a[X[i]]=1ll*W[i];
if(X[i]%2==1){
say=1;
}
if(X[i]>1){
say1=1;
}
if(X[i]==0){
sum0+=W[i];
}
else{
sum1+=W[i];
}
sum+=W[i];
}
dp[0][1]=a[0];
dp[1][0]=a[0];
dp[1][1]=a[1];
ll ans=mx(a[0],a[1]);
for(int i=2;i<N;i++){
if(i>2){
dp[i][0]=mx(dp[i-3][0]+a[i-1]+a[i-2],mx(dp[i-1][1],dp[i-1][0]));
}
else{
dp[i][0]=dp[i-1][1];
}
dp[i][1]=dp[i-1][0]+a[i];
}
ans=mx(dp[N-1][1],dp[N-1][0]);
if(say==0){
ans=sum;
}
else if(say1==0){
ans=mx(sum0,sum1);
}
return ans;
}
/*
9 9
0 0 1
1 0 4
2 0 5
3 0 2
4 0 1
5 0 5
6 0 3
7 0 2
8 0 1
*/
# | 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... |