BZOJ 2006 超级钢琴(堆+主席树)

很好的一道题。

题意：给出长度为n的数列，选择k个互不相同的区间，满足每个区间长度在[L,R]内，求所有选择的区间和的总和最大是多少。（n,k<=5e5）.

首先将区间和转化为前缀和之差，那么我们想要区间和的总和最大，一个朴素的想法是把所有满足限制的区间和排序，取最大的k个。

考虑每个右端点i，其中有效的左端点是[i-R+1,i-L+1].如果我们对于每个右端点都找到“当前”最大的区间和，那么把它们扔进大根堆里维护一下即可。

由于sum[i]一定，那么只要左端点的sum值越小越好，于是我们每次从大根堆里弹出一个数，再从弹出的这个值的右端点出发，找一个次大的值。

这样从大根堆里面弹出k次就是最大的k个区间和了。

那么我们需要维护一个数据结构支持查询静态区间第k小，显然主席树可以满足这个要求。

# include <cstdio>
# include <cstring>
# include <cstdlib>
# include <iostream>
# include <vector>
# include <queue>
# include <stack>
# include <map>
# include <bitset>
# include <set>
# include <cmath>
# include <algorithm>
using namespace std;
# define lowbit(x) ((x)&(-x))
# define pi acos(-1.0)
# define eps 1e-
# define MOD
# define INF
# define mem(a,b) memset(a,b,sizeof(a))
# define FOR(i,a,n) for(int i=a; i<=n; ++i)
# define FO(i,a,n) for(int i=a; i<n; ++i)
# define bug puts("H");
# define lch p<<,l,mid
# define rch p<<|,mid+,r
# define mp make_pair
# define pb push_back
typedef pair<int,int> PII;
typedef vector<int> VI;
# pragma comment(linker, "/STACK:1024000000,1024000000")
typedef long long LL;
int Scan() {
    int x=,f=;char ch=getchar();
    while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}
    while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}
    return x*f;
}
const int N=;
//Code begin...

int a[N], sum[N], vis[N];
struct qnode{
    int id, val;
    qnode(int _id=, int _val=):id(_id),val(_val){}
    bool operator<(const qnode &r)const{return val<r.val;}
};
priority_queue<qnode>que;
VI v;
int root[N], s[N*], ls[N*], rs[N*], sz;

void insert(int l, int r, int x, int &y, int val){
    y=++sz; s[y]=s[x]+;
    if (l==r) return ;
    ls[y]=ls[x]; rs[y]=rs[x];
    int mid=(l+r)>>;
    if (val<=mid) insert(l,mid,ls[x],ls[y],val);
    else insert(mid+,r,rs[x],rs[y],val);
}
int query(int l, int r, int x, int y, int L)
{
    if (l==r) return l;
    int mid=(l+r)>>, val=s[ls[y]]-s[ls[x]];
    if (val>=L) return query(l,mid,ls[x],ls[y],L);
    return query(mid+,r,rs[x],rs[y],L-val);
}
int main ()
{
    int n, k, L, R;
    qnode tmp;
    scanf("%d%d%d%d",&n,&k,&L,&R);
    FOR(i,,n) scanf("%d",a+i), sum[i]=sum[i-]+a[i], v.pb(sum[i]);
    v.pb();
    sort(v.begin(),v.end());
    int siz=unique(v.begin(),v.end())-v.begin();
    FOR(i,,n+) {
        int x=lower_bound(v.begin(),v.begin()+siz,sum[i-])-v.begin()+;
        insert(,siz,root[i-],root[i],x);
    }
    FOR(i,,n) {
        int l=max(i-R,), r=i-L;
        if (r<l) continue;
        int x=query(,siz,root[l],root[r+],); que.push(qnode(i,sum[i]-v[x-]));
    }
    int num=;
    LL ans=;
    while (num<k) {
        tmp=que.top(); que.pop(); ans+=tmp.val; ++vis[tmp.id]; ++num;
        int l=max(tmp.id-R,), r=tmp.id-L;
        if (r-l<vis[tmp.id]) continue;
        int x=query(,siz,root[l],root[r+],vis[tmp.id]+); que.push(qnode(tmp.id,sum[tmp.id]-v[x-]));
    }
    printf("%lld\n",ans);
    return ;
}