超级钢琴 2010年NOI

时间:2023-03-09 19:05:37
超级钢琴 2010年NOI
/*
自己yy的奇葩做法居然A了23333
不过空间好像很大 时间好像略慢.....
毕竟不是正解
前缀维护sum值 枚举区间起点
然后终点的坐标可以确定在一个范围
可持久化线段树查询区间第1大
然后放到堆里 注意每个从堆里取出来再把这个区间第2大的放进去
这里k可能减成负的 注意特判 开始wa了
还有开longlong
*/
#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
#include<algorithm>
#define pa pair<int,int>
#define mk make_pair
#define maxn 500010
using namespace std;
int n,m,l,r,k,x,root[maxn],a[maxn],s[maxn],cnt[maxn],tot;
long long ans;
struct node{
int sum,lc,rc;
}t[maxn**];
priority_queue<pa>q;
int init(){
int x=,f=;char s=getchar();
while(s<''||s>''){if(s=='-')f=-;s=getchar();}
while(s>=''&&s<=''){x=x*+s-'';s=getchar();}
return x*f;
}
int Build(int S,int L,int R){
++tot;t[tot].sum=S;
t[tot].lc=L;t[tot].rc=R;
return tot;
}
void Insert(int &root,int pre,int l,int r,int pos){
root=Build(t[pre].sum+,t[pre].lc,t[pre].rc);
if(l==r)return;
int mid=l+r>>;
if(pos<=mid)Insert(t[root].lc,t[pre].lc,l,mid,pos);
else Insert(t[root].rc,t[pre].rc,mid+,r,pos);
}
int Query(int L,int R,int l,int r,int k){
if(l==r)return l;
int mid=l+r>>;
int sum=t[t[R].lc].sum-t[t[L].lc].sum;
if(k<=sum)return Query(t[L].lc,t[R].lc,l,mid,k);
else return Query(t[L].rc,t[R].rc,mid+,r,k-sum);
}
int main()
{
freopen("piano.in","r",stdin);
//freopen("piano.out","w",stdout);
n=init();k=init();l=init();r=init();
for(int i=;i<=n;i++){
x=init();
s[i]=s[i-]+x;
a[i]=s[i];
}
int num,pos,L,R,p,len,t;
sort(a+,a++n);
num=unique(a+,a++n)-a-;
for(int i=;i<=n;i++){
pos=lower_bound(a+,a++num,s[i])-a;
Insert(root[i],root[i-],,num,pos);
}
while(!q.empty())q.pop();
for(int i=;i+l-<=n;i++){
L=i+l-,R=min(n,i+r-);
len=R-L+;++cnt[i];t=len-cnt[i]+;
pos=Query(root[L-],root[R],,num,t);
q.push(mk(a[pos]-s[i-],i));
}
for(int i=;i<=k;i++){
p=q.top().second;
ans+=q.top().first;q.pop();
L=p+l-,R=min(n,p+r-);
len=R-L+;++cnt[p];t=len-cnt[p]+;
if(t<=)continue;
pos=Query(root[L-],root[R],,num,t);
q.push(mk(a[pos]-s[p-],p));
}
cout<<ans<<endl;
return ;
}