NOI题果然难得多,不会做啊。。后来看题解有说是要主席树啥的,,理解不了。。看懂了一个RMQ+堆的。。考虑一个序列的起点i,终点的范围是可以计算的,然后维护一下前缀和sum,起点的sum一定,那么终点的sum最大是答案最优,找这个最大可以用ST算法搞定。。对于一个四元组(s,l,r,m)表示起点为s,终点在[l,r]内,最优解在终点为m取到,然后以四元组的最优解大小为key对每一个点为起点的四元组维护一个大根堆,每次取出堆顶的(s,l,r,m),累加答案,再将(s,l,m-1,m’)和(s,m+1,r,m’’)放进堆,取k次就算出答案来了。。这种思想极为巧妙。。太神了。。
#include<iostream>
#include<cstdio>
#include<cmath>
#define N 500005
#define ll long long
using namespace std;
struct node{
int s,l,r,m;
node(){};
node(int s,int l,int r,int m):s(s),l(l),r(r),m(m){};
} heap[N*2],t;
int n,k,l,r,i,j,L,R,nd=0,f[N][21];
ll ans=0ll,sum[N];
int Max(int a,int b)
{
if (sum[a]>sum[b]) return a;else return b;
}
void st()
{
int i,j;
for (j=1;j<20;j++)
for (i=1;i<=n-(1<<j)+1;i++)
f[i][j]=Max(f[i][j-1],f[i+(1<<j-1)][j-1]);
}
int qmax(int l,int r)
{
double t=(double)log(r-l+1)/(double)log(2);
int k=t;
return Max(f[l][k],f[r-(1<<k)+1][k]);
}
bool operator<(node a,node b)
{
return sum[a.m]-sum[a.s-1]<sum[b.m]-sum[b.s-1];
}
void ins(node a)
{
heap[++nd]=a;
int i=nd;
while (i>1&&heap[i/2]<heap[i]) swap(heap[i/2],heap[i]),i/=2;
}
void del()
{
heap[1]=heap[nd--];
int i=1;
while ((i*=2)<=nd)
{
if (i<nd&&heap[i]<heap[i+1]) ++i;
if (heap[i/2]<heap[i]) swap(heap[i/2],heap[i]); else break;
}
}
int main()
{
freopen("2006.in","r",stdin);
scanf("%d%d%d%d",&n,&k,&l,&r);
sum[0]=0ll;
for (i=1;i<=n;i++)
{
scanf("%I64d",&sum[i]);
sum[i]+=sum[i-1];
f[i][0]=i;
}
st();
for (i=1;i<=n;i++)
if (i+l-1>n) break;
else
{
L=i+l-1;R=min(n,i+r-1);
ins(node(i,L,R,qmax(L,R)));
}
for (i=1;i<=k;i++)
{
t=heap[1];
ans+=sum[t.m]-sum[t.s-1];
del();
if (t.m-1>=t.l) ins(node(t.s,t.l,t.m-1,qmax(t.l,t.m-1)));
if (t.r>=t.m+1) ins(node(t.s,t.m+1,t.r,qmax(t.m+1,t.r)));
}
printf("%I64d\n",ans);
}