# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
763678 | Ahmed57 | Holiday (IOI14_holiday) | C++17 | 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;
long long seg[9001] , cnt[9001];
void update(int p,int l,int r,int idx,long long val){
if(l==r){
seg[p]+=val;
cnt[p]++;
return ;
}
int md = (l+r)/2;
if(idx<=md)update(p*2,l,md,idx,val);
else update(p*2+1,md+1,r,idx,val);
cnt[p] = cnt[p*2]+cnt[p*2+1];
seg[p] = seg[p*2]+seg[p*2+1];
}
long long query(int p,int l,int r,int num){
if(l==r){
if(cnt[p]==0){
return 0;
}
return (seg[p]/cnt[p])*min(num,cnt[p]);
}
int md = (l+r)/2;
if(num>=cnt[p*2+1]){
return seg[p*2+1]+query(p*2,l,md,num-cnt[p*2+1]);
}else{
return query(p*2+1,md+1,r,num);
}
}
long long int findMaxAttraction(int n, int start, int d,
int attraction[]){
if(start==0){
priority_queue<long long> q;
long long sum = 0 , ma = 0;
for(int i = 0;i<n;i++){
q.push(-attraction[i]);
sum+=attraction[i];
while(!q.empty()&&q.size()>(d-i)){
sum+=q.top();
q.pop();
}
ma = max(ma,sum);
}
return ma;
}
map<int,int> comp,sav;
for(int i = 0;i<n;i++){
comp[attraction[i]]++;
}
int z = 0;
for(auto i:comp){
sav[i.first] = ++z;
}
long long ma = 0;
for(int i = 0;i<=start;i++){
memset(seg,0,sizeof seg);
memset(cnt,0,sizeof cnt);
for(int j = i;j<start;j++){
update(1,1,n,sav[attraction[j]],attraction[j]);
}
for(int j = start;j<n;j++){
update(1,1,n,sav[attraction[j]],attraction[j]);
long long rem = d-(min(abs(start-i),abs(start-j))*2)-max(abs(start-i),abs(start-j));
if(rem<=0)continue;
ma = max(ma,query(1,1,n,rem));
}
}
return ma;
}
/*
int main(){
int S[] = {10,2,20,30,1};
cout<<findMaxAttraction(5,2,7,S);
}*/