标签:
由于K很小,所以就直接取出最大的K个值加起来即可
考虑一个
#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<vector>
#include<queue>
#include<cmath>
#include<set>
#include<map>
#define N 500550
using namespace std;
int sc()
{
int i=0,f=1;char c=getchar();
while(c>‘9‘||c<‘0‘){if(c==‘-‘)f=-1;c=getchar();}
while(c>=‘0‘&&c<=‘9‘)i=i*10+c-‘0‘,c=getchar();
return i*f;
}
struct W{
int i,l,r,p;long long sum;
};
long long f[N][22],sum[N],ans;
int a[N],n,k,L,R;
void pre()
{
for(int i=1;i<=n;i++) f[i][0]=i;
for(int k=1;k<=19;k++)
for(int i=1;i<=n;i++)
if(i+(1<<k)>n+1)break;
else f[i][k]=sum[f[i][k-1]]>sum[f[i+(1<<k-1)][k-1]]?f[i][k-1]:f[i+(1<<k-1)][k-1];
}
int ask(int l,int r)
{
int k=log2(r-l+1);
return sum[f[l][k]]>sum[f[r-(1<<k)+1][k]]?f[l][k]:f[r-(1<<k)+1][k];
}
priority_queue<W,vector<W> > q;
bool operator<(W a,W b){return a.sum<b.sum;}
int main()
{
n=sc();k=sc(),L=sc(),R=sc();
for(int i=1;i<=n;i++)
sum[i]=sum[i-1]+(a[i]=sc());
pre();
for(int i=1;i<=n-L+1;i++)
{
int l=i+L-1,r=min(n,i+R-1);
int p=ask(l,r);
q.push((W){i,l,r,p,sum[p]-sum[i-1]});
}
while(k--)
{
W x=q.top();q.pop();ans+=x.sum;
int i=x.i,l=x.l,r=x.r,p=x.p;
if(l<p)
{
int w=ask(l,p-1);
q.push((W){i,l,p-1,w,sum[w]-sum[i-1]});
}
if(p<r)
{
int w=ask(p+1,r);
q.push((W){i,p+1,r,w,sum[w]-sum[i-1]});
}
}
cout<<ans;
return 0;
}
标签:
原文地址:http://blog.csdn.net/ws_yzy/article/details/51072474