记录编号 307444 评测结果 AAAAAAAAAA
题目名称 [NOI 2010]超级钢琴 最终得分 100
用户昵称 Gravatarkxxy 是否通过 通过
代码语言 C++ 运行时间 6.222 s
提交时间 2016-09-15 12:00:07 内存使用 84.24 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};
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];
		}
}
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;
ll ans=0;
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);
	cin>>n>>k>>L>>R;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
		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);
	}
	solve();
	cout<<ans<<endl;
	return 0;
}