bzoj 1901 主席树+树状数组

来源：互联网发布：服装货品数据分析编辑：程序博客网时间：2024/06/06 03:03

修改+查询第k小值

单纯主席树修改会打乱所有，所以再套一个树状数组维护前缀和使得修改，查询都是log

对了，bzoj上不需要读入组数，蜜汁re。。

#include<cstdio>#include<cstring>#include<iostream>#include<cmath>#include<algorithm>using namespace std;int n,m,sz,T,num_tot,num_cnt,num_l,num_r;int sum[8000005],lon[8000005],ron[8000005],num[60005];int a[50005],k[10005],p[10005],q[10005],root[60005];bool bo[10005];int L[500],R[500];int lowbit(int x){return x&(-x);}void update(int p,int &rt,int l,int r,int x,int y){    rt=++sz; sum[rt]=sum[p]+y;     lon[rt]=lon[p]; ron[rt]=ron[p];    if(l==r) return;    int mid=(l+r)/2;    if(x<=mid) update(lon[p],lon[rt],l,mid,x,y);    else update(ron[p],ron[rt],mid+1,r,x,y);}int query(int l,int r,int k){    if(l==r) return l;    int suml=0,sumr=0;    for(int i=1;i<=num_l;i++) suml+=sum[lon[L[i]]];    for(int i=1;i<=num_r;i++) sumr+=sum[lon[R[i]]];    int mid=(l+r)/2;    if(sumr-suml>=k){        for(int i=1;i<=num_l;i++) L[i]=lon[L[i]];        for(int i=1;i<=num_r;i++) R[i]=lon[R[i]];        return query(l,mid,k);    }    else{        for(int i=1;i<=num_l;i++) L[i]=ron[L[i]];        for(int i=1;i<=num_r;i++) R[i]=ron[R[i]];        return query(mid+1,r,k-(sumr-suml));    }}int main(){        char s[5];        scanf("%d%d",&n,&m);        for(int i=1;i<=n;i++){            scanf("%d",&a[i]);            num[i]=a[i];        } num_tot=n;        for(int i=1;i<=m;i++){            scanf("%s",s);            if(s[0]=='Q')                scanf("%d%d%d",&p[i],&q[i],&k[i]);            else{                scanf("%d%d",&p[i],&q[i]);                num[++num_tot]=q[i]; bo[i]=1;            }        }        sort(num+1,num+num_tot+1);        int num_cnt=unique(num+1,num+num_tot+1)-num-1;        for(int i=1;i<=n;i++){            int t=lower_bound(num+1,num+num_cnt+1,a[i])-num;            for(int j=i;j<=n;j+=lowbit(j))                update(root[j],root[j],1,num_cnt,t,1);        }        for(int i=1;i<=m;i++){            if(bo[i]){                int t=lower_bound(num+1,num+num_cnt+1,a[p[i]])-num;                for(int j=p[i];j<=n;j+=lowbit(j))                    update(root[j],root[j],1,num_cnt,t,-1);                a[p[i]]=q[i];                t=lower_bound(num+1,num+num_cnt+1,q[i])-num;                for(int j=p[i];j<=n;j+=lowbit(j))                    update(root[j],root[j],1,num_cnt,t,1);            }            else{                p[i]--; num_l=num_r=0;                for(int j=p[i];j>0;j-=lowbit(j))                    L[++num_l]=root[j];                for(int j=q[i];j>0;j-=lowbit(j))                    R[++num_r]=root[j];                printf("%d\n",num[query(1,num_cnt,k[i])]);            }        }        return 0;}

阅读全文

0 0