[NOI2010]超级钢琴 ST表+堆

个人感觉这是一道ST表不错的题,可看出是让求k个在规定长度内的区间其和的总和最大值(不要说我语文差 = =)

可以搞一个结构体 存4个量 l,from,to,r,l表示区间左端点,[from,to]为区间右端点范围,r为区间右端点。

求一下前缀和(算是预处理吧)。题目中限制长度范围为[L,R]

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<queue>
using namespace std;
const int M=500000+10;
int n,k,l,r;
int mn[M],st[M][20],val[M];
long long ans;
struct DATE{int le,fr,to,ri;};
priority_queue<DATE> pq;
inline bool operator < (DATE a,DATE b){return val[a.ri]-val[a.le-1]<val[b.ri]-val[b.le-1];}
inline int read(){
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
	return x*f;
}
inline void init(){       //ST表 
	mn[0]=-1;
	for(int i=1;i<=n;i++){
		mn[i]=(!(i&(i-1))) ? mn[i-1]+1 : mn[i-1];
		st[i][0]=i;
	}
	for(int j=1;j<=mn[n];j++)
		for(int i=1;i+(1<<j)-1<=n;i++){
			int a=st[i][j-1],b=st[i+(1<<j-1)][j-1];
			st[i][j]=(val[a]>=val[b] ? a : b);
		}
}
inline int query(int l,int r){   //返回的是下标 
	int k=mn[r-l+1];
	int a=st[l][k],b=st[r-(1<<k)+1][k];
	return val[a]>val[b] ? a : b;
}
int main(){
	n=read();k=read();l=read();r=read();
	for(int i=1;i<=n;i++){val[i]=read();val[i]+=val[i-1];}  //求出前缀 
	init();
	for(int i=1;i+l-1<=n;i++){    //大块区间push进去 
		int rig=min(i+r-1,n);
		pq.push((DATE){i,i+l-1,rig,query(i+l-1,rig)});
	}
	for(int i=1;i<=k;i++){
		DATE zhuan=pq.top();pq.pop();
		int mid=zhuan.ri;
		ans+=val[zhuan.ri]-val[zhuan.le-1];
		if(mid>zhuan.fr) pq.push((DATE){zhuan.le,zhuan.fr,mid-1,query(zhuan.fr,mid-1)});//再细找 
		if(mid<zhuan.to) pq.push((DATE){zhuan.le,mid+1,zhuan.to,query(mid+1,zhuan.to)});
	}
	printf("%lld",ans);
	return 0;
}

先从1到n-L+1循环一遍,将所有满足要求的(i,i+L-1,min(i+R-1,n),query(i+L-1,min(i+R-1,n)))push进堆中。

然后在从1循环到k,取k个较大的数(当然不是直接取了)

每次取出堆顶,一定为当前最优解ans加上

然后将[from,to]分为[from,r-1] 与 [r+1,to] push进即可。

这样就可以保证取k次即为前k大的。

(ST表中存的是下表)

代码如下:

     

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值