luogu3380/bzoj3196 二逼平衡树 (树状数组套权值线段树)

带修改区间K大值

这题有很多做法，我的做法是树状数组套权值线段树，修改查询的时候都是按着树状数组的规则找出那log(n)个线段树根，然后一起往下做

时空都是$O(nlog^2n)$的（如果离散化了的话），空间可能会被卡，但实际上点数不用开到特别大，N*200也能过

 #include<bits/stdc++.h>
 #define pa pair<int,int>
 #define CLR(a,x) memset(a,x,sizeof(a))
 using namespace std;
 typedef long long ll;
 const int maxn=5e4+,maxp=maxn*,inf=1e8;

 inline ll rd(){
     ll x=;char c=getchar();int neg=;
     while(c<''||c>''){if(c=='-') neg=-;c=getchar();}
     while(c>=''&&c<='') x=x*+c-'',c=getchar();
     return x*neg;
 }

 int root[maxn],ch[maxp][],v[maxp],pct;
 int num[maxn],tmp1[maxn],tmp2[maxn],N,M;

 inline int lowbit(int x){return x&(-x);}

 inline void update(int p){v[p]=v[ch[p][]]+v[ch[p][]];}
 inline void add(int &p,int l,int r,int x,int y){
     if(!p) p=++pct;
     if(l==r) v[p]+=y;
     else{
         int m=l+r>>;
         if(x<=m) add(ch[p][],l,m,x,y);
         else add(ch[p][],m+,r,x,y);
         update(p);
     }
 }
 inline void change(int x,int y){
     int ori=num[x];
     for(;x<=N;x+=lowbit(x)){
         if(ori!=-) add(root[x],,inf,ori,-);
         add(root[x],,inf,y,);
     }
 }

 inline void reset(int x,int y){
     for(;y;y-=lowbit(y)) tmp1[y]=root[y];
     for(;x;x-=lowbit(x)) tmp2[x]=root[x];
 }
 inline bool pushdown(int x,int y,bool b){
     bool re=;
     for(;y;y-=lowbit(y)) tmp1[y]=ch[tmp1[y]][b],re|=tmp1[y];
     for(;x;x-=lowbit(x)) tmp2[x]=ch[tmp2[x]][b],re|=tmp2[x];
     return re;
 }

 int query1(int l,int r,int x,int y,int k){
     if(k<) return ;
     if(l==r) return ;
     int m=l+r>>;
     if(k<=m){
         if(!pushdown(x,y,)) return ;
         return query1(l,m,x,y,k);
     }else{
         int w=;
         for(int i=y;i;i-=lowbit(i)) w+=v[ch[tmp1[i]][]];
         for(int i=x;i;i-=lowbit(i)) w-=v[ch[tmp2[i]][]];
         if(!pushdown(x,y,)) return w+;
         return w+query1(m+,r,x,y,k);
     }
 }
 int query2(int l,int r,int x,int y,int k){
     if(k<=) return -;
     int w=;
     for(int i=y;i;i-=lowbit(i)) w+=v[tmp1[i]];
     for(int i=x;i;i-=lowbit(i)) w-=v[tmp2[i]];
     if(w<k) return -;
     if(l==r) return l;
     int m=l+r>>;w=;
     for(int i=y;i;i-=lowbit(i)) w+=v[ch[tmp1[i]][]];
     for(int i=x;i;i-=lowbit(i)) w-=v[ch[tmp2[i]][]];
     if(k<=w){
         pushdown(x,y,);
         return query2(l,m,x,y,k);
     }else{
         pushdown(x,y,);
         return query2(m+,r,x,y,k-w);
     }
 }

 int main(){
     //freopen(".in","r",stdin);
     int i,j,k;
     N=rd(),M=rd();
     CLR(num,-);
     for(i=;i<=N;i++){
         int x=rd();
         change(i,x);num[i]=x;
     }
     for(i=;i<=M;i++){
         int a=rd(),b=rd(),c=rd();
         if(a==){
             change(b,c);num[b]=c;
         }else{
             int d=rd();
             if(a==){
                 reset(b-,c);
                 printf("%d\n",query1(,inf,b-,c,d));
             }else if(a==){
                 reset(b-,c);
                 printf("%d\n",query2(,inf,b-,c,d));
             }else if(a==){
                 reset(b-,c);
                 int rk=query1(,inf,b-,c,d);
                 reset(b-,c);
                 if(rk==) printf("-2147483647\n");
                 else printf("%d\n",query2(,inf,b-,c,rk-));
             }else if(a==){
                 reset(b-,c);
                 int rk=query1(,inf,b-,c,d+);
                 reset(b-,c);
                 int re=query2(,inf,b-,c,rk);
                 if(re==-) printf("2147483647\n");
                 else printf("%d\n",re);
             }
         }
     }
     return ;
 }