BZOJ2006: [NOI2010]超级钢琴

求和为前k大的长为L~R的区间的和

将区间和转前缀和∑Ri=Lai=s[R]−s[L−1]

右端点相同的区间放一起，以他们的区间和的最大值为代表值，把这些右端点放进一个堆里，取k次，每次取区间和最大的右端点出来，将它的左端点区间以最小的s[L-1]的L为界，取出这个L，剩下的分成两部分，求出这两部分s[L-1]的最小值，算出各自最大的区间和，把他们放进堆里

code：

#include<set>
#include<map>
#include<deque>
#include<queue>
#include<stack>
#include<cmath>
#include<ctime>
#include<bitset>
#include<string>
#include<vector>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<climits>
#include<complex>
#include<iostream>
#include<algorithm>
#define ll long long
using namespace std;

inline void swap(int &x,int &y){x^=y;y^=x;x^=y;}
inline void read(int &x)
{
char c; int f=1;
while(!((c=getchar())>='0'&&c<='9')) if(c=='-') f=-1;
x=c-'0';
while((c=getchar())>='0'&&c<='9') (x*=10)+=c-'0';
x*=f;
}
const int maxn = 510000;
const int maxb = 21;

int n,m,L,R,a[maxn];
ll s[maxn];
int mn[maxn][maxb];

int get_mn(int l,int r)
{
ll re=LLONG_MAX; int pos;
for(int i=maxb-1;i>=0;i--) if(r-(1<<i)+1>=l)
{
if(re>s[mn[r][i]]) pos=mn[r][i],re=s[pos];
r-=1<<i;
}
return pos;
}

struct node
{
ll c;
int l,r,i,p;
node(){}
node(const ll _c,int _l,int _r,int _i,int _p){c=_c;l=_l;r=_r;i=_i;p=_p;}
};
bool operator <(node x,node y){return x.c<y.c;}
priority_queue<node>Q;

int main()
{
read(n); read(m); read(L); read(R);
for(int i=1;i<=n;i++) read(a[i]),s[i]=s[i-1]+(ll)a[i];

for(int i=0;i<=n;i++) mn[i][0]=i;
for(int j=1;j<maxb;j++)
{
for(int i=1;i<=n;i++)
{
int la=i-(1<<j-1);
if(la<0) { mn[i][j]=mn[i][j-1]; continue; }
if(s[mn[i][j-1]]<s[mn[la][j-1]])
mn[i][j]=mn[i][j-1];
else mn[i][j]=mn[la][j-1];
}
}

for(int i=L;i<=n;i++)
{
int l=max(0,i-R),r=i-L;
int p=get_mn(l,r);
node tmp=node(s[i]-s[p],l,r,i,p);
Q.push(tmp);
}

ll ans=0;
while(m--)
{
node x=Q.top(); Q.pop();
ans+=x.c;
int l=x.l,r=x.p-1,p=get_mn(l,r);
if(l<=r) Q.push(node(s[x.i]-s[p],l,r,x.i,p));
l=x.p+1,r=x.r,p=get_mn(l,r);
if(l<=r) Q.push(node(s[x.i]-s[p],l,r,x.i,p));
}
printf("%lld\n",ans);

return 0;
}