Pagini recente » Cod sursa (job #838954) | Cod sursa (job #492496) | Cod sursa (job #1805317) | Cod sursa (job #1326119) | Cod sursa (job #1047137)
#include <iostream>
#include<fstream>
#include<algorithm>
using namespace std;
ifstream f("deque.in");
ofstream g("deque.out");
long heap[5000005],poz[5000005],a[5000005],nr,k;
long i,x;
long long sum;
void urca(long long &n,long long pozi)
{
while(pozi>1&& a[heap[pozi]]<a[heap[pozi/2]])
{
swap(heap[pozi],heap[pozi/2]);
poz[heap[pozi]]=pozi;
poz[heap[pozi/2]]=pozi/2;
pozi/=2;
}
}
void coboara(long long &n,long long pozi)
{
{
long long poz1=0;
while(pozi!=poz1) {
poz1=pozi;
if(2*poz1<=k && a[heap[pozi]]>a[heap[2*pozi]])
pozi=2*poz1;
if(2*poz1+1<=k && a[heap[pozi]]>a[heap[2*poz1+1]])
pozi=2*poz1+1;
swap(heap[pozi],heap[poz1]);
poz[heap[pozi]]=pozi;
poz[heap[poz1]]=poz1;
}
}}
int main()
{long long n=0,ct=1,lung;
f>>nr>>lung;
for(i=1;i<=lung;i++)
{ f>>x;
a[++n]=x;
heap[++k]=n;
poz[n]=k;
urca(n,k);
}
for(i=lung+1;i<=nr;i++)
{sum+=a[heap[1]];
a[ct]=-10000001;
urca(n,poz[ct]);
poz[heap[1]]=0;
heap[1]=heap[k--];
poz[heap[1]]=1;
coboara(n,1);
f>>x;
a[++n]=x;
heap[++k]=n;
poz[n]=k;
urca(n,k);
ct++;
}
sum+=a[heap[1]];
g<<sum;
}