记录编号 307584 评测结果 AAAAAAAAAA
题目名称 [NOI 2010]超级钢琴 最终得分 100
用户昵称 GravatarOstmbh 是否通过 通过
代码语言 C++ 运行时间 4.886 s
提交时间 2016-09-15 15:43:05 内存使用 75.82 MiB
显示代码纯文本
#include<iostream>
#include<cstdio>
#include<queue>
#include<cmath>
using namespace std;
typedef long long ll;
const ll maxn=500010;
ll n,k,L,R;
ll s[maxn]={0},a[maxn]={0},st[maxn][20]={0};
int read()
{
	int x=0,f=1;
	char c=getchar();
	while(c>'9'||c<'0')
	{
		if(c=='-')
			f=-1;
		c=getchar();
	}
	while(c<='9'&&c>='0')
	{
		x=x*10+c-'0';
		c=getchar();
	}
	return x*f;
}
inline void ST(ll a[maxn])
{
	ll m=floor(log((double)n)/log(2.0));
	for(ll i=1;i<=n;i++)
		st[i][0]=i;
	for(ll j=1;j<=m;j++)
		for(ll i=1;i<=n;i++)
		{
			if(i+(1<<j)-1>n)
				continue;
			st[i][j]=st[i][j-1];
			if(a[st[i+(1<<(j-1))][j-1]]>a[st[i][j]])
				st[i][j]=st[i+(1<<(j-1))][j-1];
		}
}
inline ll rmq(ll l,ll r)
{
	ll m=floor(log((double)r-l+1)/log(2.0));
	if(s[st[r-(1<<m)+1][m]]<s[st[l][m]])
		return st[l][m];
	else
		return st[r-(1<<m)+1][m];
}
struct Node
{
	ll ss,l,r,mx,md;
	void cal()
	{
		mx=rmq(l,r);
		md=s[mx]-s[ss-1];
	}
	bool operator <(Node b)const
	{
		return md<b.md;
	}
};
priority_queue<Node>q;
inline void init()
{
	n=read(),k=read(),L=read(),R=read();
	for(int i=1;i<=n;i++)
	{
		a[i]=read();
		s[i]=s[i-1]+a[i];
	}
	ST(s);
	Node x;
	ll lnow,rnow;
	for(int i=1;i<=n;i++)
	{
		if(i+L-1>n)
			break;
		lnow=i+L-1;
		rnow=min(i+R-1,n);
		x.ss=i,x.l=lnow,x.r=rnow;
		x.cal();
		q.push(x);
	}
}
ll ans=0;
inline void solve()
{
	Node x,y,z;
	for(int i=1;i<=k;i++)
	{
		x=q.top(),q.pop();
		ans+=x.md;
		y.ss=z.ss=x.ss;
		y.l=x.l,z.r=x.r;
		y.r=x.mx-1,z.l=x.mx+1;
		if(y.r>=y.l)
		{
			y.cal();
			q.push(y);
		}
		if(z.r>=z.l)
		{
			z.cal();
			q.push(z);
		}
	}
}
int main()
{
	freopen("piano.in","r",stdin);
	freopen("piano.out","w",stdout);
	init();
	solve();
	cout<<ans<<endl;
	return 0;
}