Acwing 2556. 第八大奇迹(树状数组套权值线段树)

题意:

有一个长度为n的序列a,一开始所有数为0,
q次操作,操作有两种:
(C x v):将a[x]修改为v
(Q l r):问[l,r]中第8大的数

数据范围:n,q<=1e5,0<=a(i)<=1e9

解法:

区间第k大,显然得用可持久化权值线段树,
因为还有修改操作,那么套一个树状数组就行了。

因为值域是[0,1e9],log(1e9)有点大,因此我将值域离散化到1e5了,log(1e5)比较小。
实测离散化之后快了很多(不离散化也能过)。

code:
#include<bits/stdc++.h>
using namespace std;
const int maxm=2e5+5;
struct QQ{
    int op,x,v;
}Q[maxm];
//
int lc[maxm*100],rc[maxm*100],cnt[maxm*100];
int rt[maxm],tot;
void update(int x,int val,int l,int r,int &k){
    if(!k)k=++tot;
    cnt[k]+=val;
    if(l==r)return ;
    int mid=(l+r)/2;
    if(x<=mid)update(x,val,l,mid,lc[k]);
    else update(x,val,mid+1,r,rc[k]);
}
//
int b[maxm],num,zero;
int a[maxm];
int n,q;
int lowbit(int i){
    return i&-i;
}
void add(int pos,int x,int val){
    for(int i=pos;i<=n;i+=lowbit(i)){
        update(x,val,1,num,rt[i]);
    }
}
//
int L[maxm],R[maxm],len1,len2;
int ask(int l,int r,int k){
    if(l==r)return l;
    int sum=0;
    for(int i=1;i<=len1;i++){
        sum-=cnt[rc[L[i]]];
    }
    for(int i=1;i<=len2;i++){
        sum+=cnt[rc[R[i]]];
    }
    int mid=(l+r)/2;
    if(sum>=k){//右边数量大于k
        for(int i=1;i<=len1;i++)L[i]=rc[L[i]];
        for(int i=1;i<=len2;i++)R[i]=rc[R[i]];
        return ask(mid+1,r,k);
    }else{//右边数量小于k
        for(int i=1;i<=len1;i++)L[i]=lc[L[i]];
        for(int i=1;i<=len2;i++)R[i]=lc[R[i]];
        return ask(l,mid,k-sum);
    }
}
int kth(int l,int r){
    if(r-l+1<8)return zero;
    len1=len2=0;
    for(int i=l-1;i;i-=lowbit(i))L[++len1]=rt[i];
    for(int i=r;i;i-=lowbit(i))R[++len2]=rt[i];
    return ask(1,num,8);
}
//
signed main(){
    scanf("%d%d",&n,&q);
    //离线
    b[++num]=0;
    for(int i=1;i<=q;i++){
        char op[3];scanf("%s",op);
        if(op[0]=='C')Q[i].op=1;
        else Q[i].op=2;
        scanf("%d%d",&Q[i].x,&Q[i].v);
        if(Q[i].op==1)b[++num]=Q[i].v;
    }
    //离散化
    sort(b+1,b+1+num);
    num=unique(b+1,b+1+num)-b-1;
    for(int i=1;i<=q;i++){
        if(Q[i].op==1){
            Q[i].v=lower_bound(b+1,b+1+num,Q[i].v)-b;
        }
    }
    //插入初始值0
    zero=lower_bound(b+1,b+1+num,0)-b;
    for(int i=1;i<=n;i++){
        a[i]=zero;
        add(i,a[i],1);
    }
    //处理询问
    for(int i=1;i<=q;i++){
        if(Q[i].op==1){//修改
            int x=Q[i].x,v=Q[i].v;
            add(x,a[x],-1);
            a[x]=v;
            add(x,a[x],1);
        }else{//查询
            int l=Q[i].x,r=Q[i].v;
            int ans=kth(l,r);
            printf("%d\n",b[ans]);
        }
    }
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值