2006: [NOI2010]超级钢琴

RMQ+Heap+区间分裂。

对于任意区间右端点i，其左端点取值在l,r之间，若左端点为m，则v为max(sum[i]-sum[m-1])，显然这里i是不变的，所以可以用rmq查询m的位置，然后计算v。

现将所有右端点扫一遍，然后扔到堆里面，堆中节点记录的是决策，即右端点i，左端点区间，优先级由v决定。

然后取出堆顶，v加到ans里去，分裂[l,r]为[l,m-1]和[m+1,r]，rmq出新的v，再将分裂后的区间加到堆里。

重复k次就好了。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
const int N=500000+5;
typedef long long ll;
struct Heapnode{
	int i,l,r,v,m;
	bool operator<(const Heapnode &rhs)const{
		return v<rhs.v;
	}
};
int n,k,L,R,st[N][21],sum[N];
void rmq_init(){
	for(int i=0;i<=n;i++)st[i][0]=i;
	for(int j=1;(1<<j)<=n;j++)
	for(int i=0;i+(1<<j)-1<=n;i++){
		int l=st[i][j-1],r=st[i+(1<<(j-1))][j-1];
		if(sum[l]<sum[r])st[i][j]=l;
		else st[i][j]=r;
	}
}
int rmq(int l,int r){
	l--;r--;
	int k=0;
	while((1<<(k+1))<=r-l+1)k++;
	return sum[st[l][k]]<sum[st[r-(1<<k)+1][k]]?st[l][k]:st[r-(1<<k)+1][k];
}
int main(){
	scanf("%d%d%d%d",&n,&k,&L,&R);
	for(int i=1;i<=n;i++)scanf("%d",&sum[i]);
	for(int i=1;i<=n;i++)sum[i]+=sum[i-1];
	rmq_init();
	priority_queue<Heapnode>q;
	ll ans=0;
	int l,r,v,m,i;
	for(i=1;i<=n;i++){
		l=i-R+1;r=i-L+1;
		l=max(l,1);r=max(r,1);
		if(i-l+1<L)continue;
		m=rmq(l,r);m++;v=sum[i]-sum[m-1];
		q.push((Heapnode){i,l,r,v,m});
	}
	while(k--){
		Heapnode x=q.top();q.pop();
		ans+=x.v;
		i=x.i;l=x.l;r=x.r;m=x.m;
		int tmp;
		if(l<m){
			tmp=rmq(l,m-1);tmp++;
			v=sum[i]-sum[tmp-1];
			q.push((Heapnode){i,l,m-1,v,tmp});
		}
		if(m<r){
			tmp=rmq(m+1,r);tmp++;
			v=sum[i]-sum[tmp-1];
			q.push((Heapnode){i,m+1,r,v,tmp});
		}
	}
	printf("%lld",ans);
	return 0;
}