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<bits/stdc++.h>
#pragma GCC optimize("Ofast")
#pragma GCC target("avx2")
using namespace std;
#define F first
#define S second
#define ll long long
// #define int ll
#define pb push_back
#define sz(s) (int)s.size()
#define pii pair<int,int>
#define all(v) v.begin(),v.end()
#define mem(a,i) memset(a,i,sizeof(a))
#define in insert
#define lb lower_bound
#define ub upper_bound
#define y1 yy
#define ppb pop_back
#define ull unsigned ll
const int MAX=5e5+55;
const int inf=1e9;
const int N=2e5;
const int C=331;
const int C1=431;
const int mod=1e9+9;
const int mod1=1e9+9;
#include "holiday.h"
ll dpL[MAX],dpR[MAX];
ll dp1L[MAX],dp1R[MAX];
ll findMaxAttraction(int n, int start, int d, int attraction[]) {
for(ll ds=0;ds<=d;ds++){
multiset<ll> st;
ll ans=0;
ll cur=0;
for(ll i=start+1;i<n;i++){
st.in(attraction[i]);
cur+=attraction[i];
while(!st.empty()&&sz(st)>ds-(i-start)){
cur-=*st.begin();
st.erase(st.begin());
}
ans=max(ans,cur);
}
dpR[ds]=ans;
st.clear();
ans=cur=0;
for(ll i=start;i<n;i++){
st.in(attraction[i]);
cur+=attraction[i];
while(!st.empty()&&sz(st)>ds-2*(i-start)){
cur-=*st.begin();
st.erase(st.begin());
}
ans=max(ans,cur);
}
dp1R[ds]=ans;
}
for(ll ds=0;ds<=d;ds++){
multiset<ll> st;
ll ans=0;
ll cur=0;
for(ll i=start-1;i>=0;i--){
st.in(attraction[i]);
cur+=attraction[i];
while(!st.empty()&&sz(st)>ds-(start-i-1)){
cur-=*st.begin();
st.erase(st.begin());
}
ans=max(ans,cur);
}
dpL[ds]=ans;
st.clear();
ans=cur=0;
for(ll i=start;i>=0;i--){
st.in(attraction[i]);
cur+=attraction[i];
while(!st.empty()&&sz(st)>ds-2*(start-i-1)){
cur-=*st.begin();
st.erase(st.begin());
}
ans=max(ans,cur);
}
dp1L[ds]=ans;
}
// cout<<dpR[4]<<"\n";
ll ans=0;
for(ll i=0;i<=d;i++){
ans=max(ans,dpR[i]+dp1L[d-i-1]);
ans=max(ans,dp1R[i]+dpL[d-i-1]);
}
return ans;
}
# | 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... |