bzoj 3110 [Zjoi2013]K大数查询（树套树）

  1 #include<cstdio>
  2 #include<cstring>
  3 #include<iostream>
  4 #include<algorithm>
  5 #define FOR(a,b,c) for(int a=(b);a<=(c);a++)
  6 using namespace std;
  7 
  8 typedef long long ll;
  9 const int N = 200000+10;
 10 const int M = 20000000+10;
 11 
 12 struct Tnode{
 13     int lc,rc,add,sum;
 14     Tnode(){}
 15 }T[M];
 16 
 17 int read() {
 18     char c=getchar();
 19     int f=1,x=0;
 20     while(!isdigit(c)) {
 21         if(c==‘-‘) f=-1; c=getchar();
 22     }
 23     while(isdigit(c))
 24         x=x*10+c-‘0‘,c=getchar();
 25     return x*f;
 26 }
 27 
 28 int n,m,sz;
 29 int rt[M];
 30 
 31 void pushdown(int u,int l,int r)
 32 {
 33     if(!T[u].add || l==r) return ;
 34     if(!T[u].lc) T[u].lc=++sz;
 35     if(!T[u].rc) T[u].rc=++sz;
 36     int v=T[u].add , mid=(l+r)>>1;
 37     T[T[u].lc].add+=v;
 38     T[T[u].rc].add+=v;
 39     T[T[u].lc].sum+=v*(mid-l+1);
 40     T[T[u].rc].sum+=v*(r-mid);
 41     T[u].add=0;
 42 }
 43 void update(int &u,int l,int r,int L,int R)
 44 {
 45     if(!u) u=++sz;
 46     pushdown(u,l,r);
 47     if(L<=l&&r<=R) {
 48         T[u].add++;
 49         T[u].sum+=r-l+1;
 50     } else {
 51         int mid=(l+r)>>1;
 52         if(L<=mid) update(T[u].lc,l,mid,L,R);
 53         if(mid<R ) update(T[u].rc,mid+1,r,L,R);
 54         T[u].sum=T[T[u].lc].sum+T[T[u].rc].sum;
 55     }
 56 }
 57 int query(int u,int l,int r,int L,int R)
 58 {
 59     if(!u) return 0;
 60     pushdown(u,l,r);
 61     if(L<=l&&r<=R) return T[u].sum;
 62     else {
 63         int mid=(l+r)>>1,ans=0;
 64         if(L<=mid) ans+=query(T[u].lc,l,mid,L,R);
 65         if(mid<R ) ans+=query(T[u].rc,mid+1,r,L,R);
 66         return ans;
 67     }
 68 }
 69 void change(int a,int b,int c)
 70 {
 71     int u=1,l=1,r=n;
 72     while(l!=r) {
 73         int mid=(l+r)>>1;
 74         update(rt[u],1,n,a,b);
 75         if(c<=mid) r=mid,u=u<<1;
 76         else l=mid+1,u=u<<1|1;
 77     }
 78     update(rt[u],1,n,a,b);
 79 }
 80 int query(int a,int b,int c)
 81 {
 82     int u=1,l=1,r=n;
 83     while(l!=r) {
 84         int mid=(l+r)>>1;
 85         int t=query(rt[u<<1],1,n,a,b);
 86         if(c<=t) r=mid,u=u<<1;
 87         else l=mid+1,u=u<<1|1,c-=t;
 88     }
 89     return l;
 90 }
 91 
 92 int main()
 93 {
 94     //freopen("in.in","r",stdin);
 95     //freopen("out.out","w",stdout);
 96     n=read(),m=read();
 97     int op,a,b,c;
 98     FOR(i,1,m) {
 99         op=read(),a=read(),b=read(),c=read();
100         if(op==1) {
101             change(a,b,n-c+1);
102         } else {
103             printf("%d\n",n-query(a,b,c)+1);
104         }
105     }
106     return 0;
107 }