[BZOJ2006][NOI2010]超级钢琴(st表+堆贪心)

来源:互联网 发布:线切割控制编程系统 编辑:程序博客网 时间:2024/05/15 23:47

题目:

我是超链接

题解:

一个区间里的所有权值和要用前缀和相减表示啊,预处理出前缀和的区间最大值
目的O(1)查询给定范围内的最大值及其位置,固定一个前缀和最小的地方再去[l,r]区间里找一个前缀和最大的位置
我们就得到了以i为起点,[L,R]范围内的最优值及其位置,维护位置看起来很复杂,实际上只需要给st表加一个结构体就好了
把那些东西push进堆里去,弹出来第一个点时,ta的最优值的位置肯定要废掉,那就重新取最优值,为了避开maxlocation(loc),就把原来的区间分成两个区间[L,loc-1]和[loc+1,R],再找到最优位置和最优值push进去
选取k次就是答案了啊
不是所有的点左右边界都是[L,R]啊!!!区间还要分呢,还需要维护左右边界值呢

代码:

#include <queue>#include <cmath>#include <cstdio>#include <cstring>#include <iostream>#define LL long long #define sz 20#define N 500005using namespace std;struct maxn{LL z;int loc;}maxx[N][sz];struct hh{int id;LL z;int loc,lm,rm;};bool operator <(const hh &a,const hh &b){return a.z<b.z;} priority_queue<hh>q;int a[N],n;LL sum[N],ans;void st(){    int i,j;    for (j=1;j<sz;j++)      for (i=1;i<=n;i++)        if (i+(1<<j)-1<=n)        {            if (maxx[i][j-1].z<maxx[i+(1<<j-1)][j-1].z)              maxx[i][j]=maxx[i+(1<<j-1)][j-1];            else              maxx[i][j]=maxx[i][j-1];        }}int main(){    int kk,ll,rr,i,lm,rm;    scanf("%d%d%d%d",&n,&kk,&ll,&rr);    for (i=1;i<=n;i++) scanf("%d",&a[i]),sum[i]=sum[i-1]+a[i];    for (i=1;i<=n;i++) maxx[i][0].z=sum[i],maxx[i][0].loc=i;    st();    for (i=1;i<=n;i++)    {        int l=i+ll-1,r=i+rr-1;        lm=l;rm=min(r,n);r=rm;        if (l>r) break;        int k=log2(r-l+1),loc,Max;        if (maxx[l][k].z<maxx[r-(1<<k)+1][k].z)          loc=maxx[r-(1<<k)+1][k].loc,Max=maxx[r-(1<<k)+1][k].z;        else loc=maxx[l][k].loc,Max=maxx[l][k].z;        q.push((hh){i,Max-sum[i-1],loc,lm,rm});         }    for (i=1;i<=kk;i++)    {        hh now=q.top(); q.pop();        ans+=now.z;        int k,loc,Max;        int l=now.lm,r=now.loc-1;           if (l<=r)        {            k=log2(r-l+1);            if (maxx[l][k].z<maxx[r-(1<<k)+1][k].z)              loc=maxx[r-(1<<k)+1][k].loc,Max=maxx[r-(1<<k)+1][k].z;            else loc=maxx[l][k].loc,Max=maxx[l][k].z;            q.push((hh){now.id,Max-sum[now.id-1],loc,now.lm,r});        }           l=now.loc+1,r=now.rm;        if (l<=r)        {        k=log2(r-l+1);        if (maxx[l][k].z<maxx[r-(1<<k)+1][k].z)          loc=maxx[r-(1<<k)+1][k].loc,Max=maxx[r-(1<<k)+1][k].z;        else loc=maxx[l][k].loc,Max=maxx[l][k].z;        q.push((hh){now.id,Max-sum[now.id-1],loc,l,now.rm});            }    }    printf("%lld",ans);}
原创粉丝点击