# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
253350 | eohomegrownapps | Holiday (IOI14_holiday) | C++14 | 0 ms | 0 KiB |
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"holiday.h"
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll findMaxFrom(int n, int start, int d, int attraction[]){
//from start
int numwalk = d/2;
int numtake = d-numwalk;
priority_queue<ll,vector<ll>,greater<ll>> pq;
ll tot = 0;
ll ans = 0;
for (int i = start; i<=min(n-1,start+numwalk); i++){
tot+=attraction[i];
pq.push(attraction[i]);
}
if (numtake==numwalk&&start+numwalk<=n-1){
tot-=pq.top();
pq.pop();
}
ans=max(ans,tot);
int cur = start+numwalk;
while (cur<n-1&&numtake>0&&numwalk<d){
//walk one more
numtake--;
numwalk++;
cur++;
tot+=attraction[cur];
pq.push(attraction[cur]);
tot-=pq.top();
pq.pop();
tot-=pq.top();
pq.pop();
ans=max(ans,tot);
}
return ans;
}
ll findMaxAttraction(int n, int start, int d, int attraction[]) {
ll mx = 0;
for (int i = max(0,start-d); i<=start; i++){
mx=max(mx,findMaxFrom(n,i,d-(start-i),attraction));
}
if (i>5000){
return mx;
}
reverse(attraction,attraction+n);
start=n-1-start;
for (int i = max(0,start-d); i<=start; i++){
mx=max(mx,findMaxFrom(n,i,d-(start-i),attraction));
}
return mx;
}