BZOJ2006: [NOI2010]超级钢琴

BZOJ2006: [NOI2010]超级钢琴

优先队列·RMQ

题解：

http://blog.csdn.net/orpinex/article/details/7012836

这道题合法的子序列是非常多的，如果朴素显然是无法做出这道题
有一个非常美妙的想法，对于给定的起点，起点的权值已知了，子序列的个数是确定的
那么记录一下前缀和s[i]，对于给定的起点，实际上就是询问起点所代表的那一段区间的最大权值，这个就是RMQ问题，ST算法可以在O(nlogn)-O(1)的时间完成
那么对于每个点i，有一个pos[i]指向i所对应的区间里s[j]的最大值
用堆来维护一个五元组(st,idx,l,r,v)，st代表起点，idx代表起点对应的区间里最大值的下标，l,r代表对应的区间，v代表这个子序列的值
那么我们每次从堆中取出这样一个五元组以后，要将区间分裂，因为i-j这个区间不能取了，所以要将(st,idx,l,r,v)分裂为(st,idx’,l,idx-1,v),(st,idx”,idx+1,r,v)两个区间
由于最多取k个，堆中最多有n+k个元素，时间复杂度为O((n+k)log(n+k))

由于ans没用long long，WA了一发。。。

Code：

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <queue>
#define D(x) cout<<#x<<" = "<<x<<"  "
#define E cout<<endl
using namespace std;
typedef pair<int,int> pii;
const int N = 500005;

int n,K,L,R,a[N],s[N];
long long ans;

struct RMQ{
    pii d[N<<1][20];
    void init(int a[]){
        for(int i=1;i<=n;i++){ d[i][0].first=a[i]; d[i][0].second=i; }
        for(int j=1;j<=19;j++){ for(int i=1;i<=n;i++){ d[i][j]=max(d[i][j-1],d[i+(1<<(j-1))][j-1]); } }
    }
    pii query(int l,int r){
        int i=19; while((1<<i) > (r-l+1)) i--;
        return max(d[l][i],d[r-(1<<i)+1][i]);
    }
} st;

struct Data{
    int a,lb,rb,pos,val;
    Data(int _a,int _lb,int _rb,int _pos,int _val){ a=_a; lb=_lb; rb=_rb; pos=_pos; val=_val; }
    bool operator < (const Data &other) const { return val < other.val; }
    void debug(){ puts("Data"); D(a); D(lb); D(rb); D(pos); D(val); E; }
};

priority_queue<Data> q;

void insert(int a,int lb,int rb){
//  D(a); D(lb); D(rb); E;
    if(rb < lb) return;
    pii x=st.query(lb,rb);
//  D(x.second); E;
    q.push(Data(a,lb,rb,x.second,s[x.second]-s[a-1]));
}

int main(){
    freopen("a.in","r",stdin);
    scanf("%d%d%d%d",&n,&K,&L,&R);
    for(int i=1;i<=n;i++) scanf("%d",a+i);
    for(int i=1;i<=n;i++) s[i]=s[i-1]+a[i];
    st.init(s);
    for(int i=1;i<=n-L+1;i++) insert(i,i+L-1,min(i+R-1,n));
    while(K--){
        Data x=q.top(); q.pop(); 
//      x.debug();
        ans+=x.val;
        insert(x.a,x.lb,x.pos-1);
        insert(x.a,x.pos+1,x.rb);
    }
    printf("%lld\n",ans);
}