BZOJ 3110, K大数查询

Problem

传送门

Mean

要求维护一个数列，支持在某部分的每个位置填上一个数以及查询某部分第K大的数字。

Analysis

终于学习了一下树套树的姿势……其实和自己想的也差不多……但是不学还是打不出代码来……
外层是权值线段树，内层是区间线段树。
要注意的细节比较多。二分查询过程中结果可能超过 Maxlongint ，需要用 unsigned int 类型传递；权值可以翻转后整体右移等等。
但最主要的还是实现树套树啦~~~

Code

#include<cstdio>
const int N=50000,L=100001,M=16000005;
int n,m,cmd,a,b,c,cnt,root[M],ls[M],rs[M];
unsigned int sum[M],tag[M];
void read(int &x){
    char c;
    bool p=0;
    while((c=getchar())<'0' || c>'9') if(c=='-') p=1;
    x=c-'0';
    while((c=getchar())>='0' && c<='9') x=x*10+c-'0';
    if(p) x=-x;
}
void down(int o,int l,int r){
    if(!ls[o]) ls[o]=++cnt;
    if(!rs[o]) rs[o]=++cnt;
    int mid=l+r>>1;
    tag[ls[o]]+=tag[o],tag[rs[o]]+=tag[o];
    sum[ls[o]]+=(mid-l+1)*tag[o],sum[rs[o]]+=(r-mid)*tag[o];
    tag[o]=0;
}
void modify(int &o,int l,int r,int ql,int qr){
    if(!o) o=++cnt;
    if(tag[o] && l<r) down(o,l,r);
    if(ql==l && r==qr){
        sum[o]+=r-l+1;
        tag[o]++;
        return;
    }
    int mid=l+r>>1;
    if(qr<=mid) modify(ls[o],l,mid,ql,qr);
    else if(ql>mid) modify(rs[o],mid+1,r,ql,qr);
    else{
        modify(ls[o],l,mid,ql,mid);
        modify(rs[o],mid+1,r,mid+1,qr);
    }
    sum[o]=sum[ls[o]]+sum[rs[o]];
}
void insert(int o,int l,int r,int a,int b,int q){
    while(l<r){
        int mid=l+r>>1;
        modify(root[o],1,n,a,b);
        if(q<=mid) r=mid,o<<=1;
        else l=mid+1,o=o<<1|1;
    }
    modify(root[o],1,n,a,b);
}
unsigned int find(int o,int l,int r,int ql,int qr){
    if(!o) return 0;
    if(l==ql && r==qr) return sum[o];
    if(tag[o]) down(o,l,r);
    int mid=l+r>>1;
    if(qr<=mid) return find(ls[o],l,mid,ql,qr);
    if(ql>mid) return find(rs[o],mid+1,r,ql,qr);
    return find(ls[o],l,mid,ql,mid)+find(rs[o],mid+1,r,mid+1,qr);
}
int query(int o,int l,int r,int a,int b,unsigned int c){
    while(l<r){
        int mid=l+r>>1;
        unsigned int t=find(root[o<<1],1,n,a,b);
        if(t<c) l=mid+1,c-=t,o=o<<1|1;
        else r=mid,o<<=1;
    }
    return l;
}
int main(){
    read(n),read(m);
    while(m--){
        read(cmd),read(a),read(b),read(c);
        if(cmd==1) insert(1,1,L,a,b,N-c+1);
        else printf("%d\n",N+1-query(1,1,L,a,b,c));
    }
    return 0;
}