【NOI2010】超级钢琴

22 篇文章 0 订阅
4 篇文章 0 订阅

题目大意

给出一个长度为n的序列,求所有长度在L到R之间的连续子序列的权值和的前k大的和

解法

首先将以每个点为开头的长度在L到R之间的序列的权值和最大的区间扔进堆里,每次拿出堆顶,然后再放入以当前点为开头的次大,如此类推,就可解决该问题了。
那么如何维护k大,下面介绍几种方法:
1、可持续化treap,每次移动一个点,相当于是一个框在移动,即加入一个,删除一个,每次直接查找就好了
2、可持续化线段树(主席树),既然是区间内的第k大,很明显可以用主席树来找,每次也是直接找就好了
3、RMQ,维护一个区间的最大值,但是与前面方法的处理上有稍稍的不一样,每次那出堆顶之后,不是找次大,而是将这个区间分裂成两个,那么次大一定会在其中一个区间里(最简短的方法吼吼吼)

#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<iostream>

#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)

using namespace std;

typedef long long LL;
typedef double db;

int get(){
    char ch;
    int s=0;
    bool pd=0;
    while(ch=getchar(),(ch<'0'||ch>'9')&&ch!='-');
    if (ch=='-')pd=1;
    else s=ch-'0';
    while(ch=getchar(),ch>='0'&&ch<='9')s=s*10+ch-'0';
    if (pd)return -s;
    return s;
}

const int N = 500010;

struct element{
    int l,r,w,v,st;
}a[N*2];
int q[N],rmq[N][20],k,n,l,r,m;
LL ans;

int getw(int l,int r){
    int t=log(r-l+1)/log(2);
    if (q[rmq[l][t]]>q[rmq[r-(1<<t)+1][t]])return rmq[l][t];
    return rmq[r-(1<<t)+1][t];
}

void insert(int st,int l,int r){
    a[++m].st=st;
    a[m].l=l;a[m].r=r;
    a[m].v=q[a[m].w=getw(l,r)]-q[st-1];
    int now=m;
    while(now>1&&a[now].v>a[now/2].v){
        swap(a[now],a[now/2]);
        now/=2;
    }
}

void del(){
    a[1]=a[m--];
    int now=1;
    while(now*2<=m){
        int x=now*2+1;
        if (now*2==m||a[x-1].v>a[x].v)x--;
        if (a[x].v<a[now].v)break;
        swap(a[x],a[now]);
        now=x;
    }
}

int main(){
    freopen("piano.in","r",stdin);
    freopen("piano.out","w",stdout);
    n=get();k=get();l=get();r=get();
    fo(i,1,n)q[i]=q[i-1]+get();
    int tim=log(n)/log(2);
    fo(i,1,n)rmq[i][0]=i;
    fo(j,1,tim)
        fo(i,1,n-(1<<j)+1)
        if (q[rmq[i][j-1]]>q[rmq[i+(1<<(j-1))][j-1]])rmq[i][j]=rmq[i][j-1];
        else rmq[i][j]=rmq[i+(1<<j-1)][j-1];
    fo(i,1,n-l+1)insert(i,i+l-1,min(i+r-1,n));
    while(k){
        k--;
        element x=a[1];
        del();
        ans+=x.v;
        if (x.w>x.l)insert(x.st,x.l,x.w-1);
        if (x.w<x.r)insert(x.st,x.w+1,x.r);
    }
    printf("%lld\n",ans);
    fclose(stdin);
    fclose(stdout);
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值