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[]) {
int begprefix = start;
int endprefix = n-1-start;
if (endprefix<begprefix){
reverse(attraction,attraction+n);
swap(begprefix,endprefix);
}
ll mx = 0;
for (int i = 0; i<=start; i++){
mx=max(mx,findMaxFrom(n,i,d,attraction));
}
return mx;
}
# | 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... |