题目:
题解:
一个区间里的所有权值和要用前缀和相减表示啊,预处理出前缀和的区间最大值
目的O(1)查询给定范围内的最大值及其位置,固定一个前缀和最小的地方再去[l,r]区间里找一个前缀和最大的位置
我们就得到了以i为起点,[L,R]范围内的最优值及其位置,维护位置看起来很复杂,实际上只需要给st表加一个结构体就好了
把那些东西push进堆里去,弹出来第一个点时,ta的最优值的位置肯定要废掉,那就重新取最优值,为了避开maxlocation(loc),就把原来的区间分成两个区间[L,loc-1]和[loc+1,R],再找到最优位置和最优值push进去
选取k次就是答案了啊
不是所有的点左右边界都是[L,R]啊!!!区间还要分呢,还需要维护左右边界值呢
代码:
#include <queue>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#define LL long long
#define sz 20
#define N 500005
using namespace std;
struct maxn{LL z;int loc;}maxx[N][sz];
struct hh{int id;LL z;int loc,lm,rm;};
bool operator <(const hh &a,const hh &b){return a.z<b.z;}
priority_queue<hh>q;
int a[N],n;LL sum[N],ans;
void st()
{
int i,j;
for (j=1;j<sz;j++)
for (i=1;i<=n;i++)
if (i+(1<<j)-1<=n)
{
if (maxx[i][j-1].z<maxx[i+(1<<j-1)][j-1].z)
maxx[i][j]=maxx[i+(1<<j-1)][j-1];
else
maxx[i][j]=maxx[i][j-1];
}
}
int main()
{
int kk,ll,rr,i,lm,rm;
scanf("%d%d%d%d",&n,&kk,&ll,&rr);
for (i=1;i<=n;i++) scanf("%d",&a[i]),sum[i]=sum[i-1]+a[i];
for (i=1;i<=n;i++) maxx[i][0].z=sum[i],maxx[i][0].loc=i;
st();
for (i=1;i<=n;i++)
{
int l=i+ll-1,r=i+rr-1;
lm=l;rm=min(r,n);r=rm;
if (l>r) break;
int k=log2(r-l+1),loc,Max;
if (maxx[l][k].z<maxx[r-(1<<k)+1][k].z)
loc=maxx[r-(1<<k)+1][k].loc,Max=maxx[r-(1<<k)+1][k].z;
else loc=maxx[l][k].loc,Max=maxx[l][k].z;
q.push((hh){i,Max-sum[i-1],loc,lm,rm});
}
for (i=1;i<=kk;i++)
{
hh now=q.top(); q.pop();
ans+=now.z;
int k,loc,Max;
int l=now.lm,r=now.loc-1;
if (l<=r)
{
k=log2(r-l+1);
if (maxx[l][k].z<maxx[r-(1<<k)+1][k].z)
loc=maxx[r-(1<<k)+1][k].loc,Max=maxx[r-(1<<k)+1][k].z;
else loc=maxx[l][k].loc,Max=maxx[l][k].z;
q.push((hh){now.id,Max-sum[now.id-1],loc,now.lm,r});
}
l=now.loc+1,r=now.rm;
if (l<=r)
{
k=log2(r-l+1);
if (maxx[l][k].z<maxx[r-(1<<k)+1][k].z)
loc=maxx[r-(1<<k)+1][k].loc,Max=maxx[r-(1<<k)+1][k].z;
else loc=maxx[l][k].loc,Max=maxx[l][k].z;
q.push((hh){now.id,Max-sum[now.id-1],loc,l,now.rm});
}
}
printf("%lld",ans);
}