题目大意
给出一个长度为n的序列,求所有长度在L到R之间的连续子序列的权值和的前k大的和
解法
首先将以每个点为开头的长度在L到R之间的序列的权值和最大的区间扔进堆里,每次拿出堆顶,然后再放入以当前点为开头的次大,如此类推,就可解决该问题了。
那么如何维护k大,下面介绍几种方法:
1、可持续化treap,每次移动一个点,相当于是一个框在移动,即加入一个,删除一个,每次直接查找就好了
2、可持续化线段树(主席树),既然是区间内的第k大,很明显可以用主席树来找,每次也是直接找就好了
3、RMQ,维护一个区间的最大值,但是与前面方法的处理上有稍稍的不一样,每次那出堆顶之后,不是找次大,而是将这个区间分裂成两个,那么次大一定会在其中一个区间里(最简短的方法吼吼吼)
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<iostream>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
typedef long long LL;
typedef double db;
int get(){
char ch;
int s=0;
bool pd=0;
while(ch=getchar(),(ch<'0'||ch>'9')&&ch!='-');
if (ch=='-')pd=1;
else s=ch-'0';
while(ch=getchar(),ch>='0'&&ch<='9')s=s*10+ch-'0';
if (pd)return -s;
return s;
}
const int N = 500010;
struct element{
int l,r,w,v,st;
}a[N*2];
int q[N],rmq[N][20],k,n,l,r,m;
LL ans;
int getw(int l,int r){
int t=log(r-l+1)/log(2);
if (q[rmq[l][t]]>q[rmq[r-(1<<t)+1][t]])return rmq[l][t];
return rmq[r-(1<<t)+1][t];
}
void insert(int st,int l,int r){
a[++m].st=st;
a[m].l=l;a[m].r=r;
a[m].v=q[a[m].w=getw(l,r)]-q[st-1];
int now=m;
while(now>1&&a[now].v>a[now/2].v){
swap(a[now],a[now/2]);
now/=2;
}
}
void del(){
a[1]=a[m--];
int now=1;
while(now*2<=m){
int x=now*2+1;
if (now*2==m||a[x-1].v>a[x].v)x--;
if (a[x].v<a[now].v)break;
swap(a[x],a[now]);
now=x;
}
}
int main(){
freopen("piano.in","r",stdin);
freopen("piano.out","w",stdout);
n=get();k=get();l=get();r=get();
fo(i,1,n)q[i]=q[i-1]+get();
int tim=log(n)/log(2);
fo(i,1,n)rmq[i][0]=i;
fo(j,1,tim)
fo(i,1,n-(1<<j)+1)
if (q[rmq[i][j-1]]>q[rmq[i+(1<<(j-1))][j-1]])rmq[i][j]=rmq[i][j-1];
else rmq[i][j]=rmq[i+(1<<j-1)][j-1];
fo(i,1,n-l+1)insert(i,i+l-1,min(i+r-1,n));
while(k){
k--;
element x=a[1];
del();
ans+=x.v;
if (x.w>x.l)insert(x.st,x.l,x.w-1);
if (x.w<x.r)insert(x.st,x.w+1,x.r);
}
printf("%lld\n",ans);
fclose(stdin);
fclose(stdout);
return 0;
}