P3332 [ZJOI2013]K大数查询

区间更新，区间查询
区间更新用的是差分的做法实现的，理论时间复杂度应该是最快的？
不明白为什么有大佬能跑到几百ms，好像用的是整体二分。。。

详解以后有时间单独贴，理解起来不是很容易

差分做法：

#include<bits/stdc++.h>
#define ls ls_[i]
#define rs rs_[i]
#define mid (l+r)/2
using namespace std;
typedef long long ll;

const int NN=100100;
ll n,m,k,x,y,z,q,W,T,N,S,op,cnt,tmp,dst,cas,sum,val,tot,idx,test;

ll c[NN*40],d[NN*40];
int ls_[NN*40],rs_[NN*40];
int rot[NN],now_rotx[NN],now_roty[NN];

void update(int l,int r,int pos,int val,int x,int &i){
    if(!i) i=++tot;
    c[i]+=val;d[i]+=val*x;
    if(l==r) return ;
    if(mid>=pos) update(l,mid,pos,val,x,ls);
    else update(mid+1,r,pos,val,x,rs);
}

#define rx now_rotx[i]
#define ry now_roty[i]

int main(){
    ios::sync_with_stdio(false);
    N=NN;
    cin>>n>>m;
    while(m--){
        cin>>op>>x>>y>>k;
        if(op==1){
            y++;
            for(int j=x;j<=n;j+=j&-j)
                update(0,N,k,1,x-1,rot[j]);
            for(int j=y;j<=n;j+=j&-j)
                update(0,N,k,-1,y-1,rot[j]);
        }
        else{
            int l=0,r=N;x--;
            for(int i=y;i;i-=i&-i) ry=rot[i];
            for(int i=x;i;i-=i&-i) rx=rot[i];
            //查询的非递归写法，递归式的我没想到该怎么写
            //因为树状数组节点的有序性决定它只适合做外壳
            //也许记录每个lowbit位置存前向星能放进内层？可以一试
            while(l!=r){
                ll c_rs=0;
                for(int i=y;i;i-=i&-i){
                    c_rs+=c[rs_[ry]]*y;
                    c_rs-=d[rs_[ry]];
                }
                for(int i=x;i;i-=i&-i){
                    c_rs-=c[rs_[rx]]*x;
                    c_rs+=d[rs_[rx]];
                }
                if(c_rs>=k){
                    l=mid+1;
                    for(int i=y;i;i-=i&-i) ry=rs_[ry];
                    for(int i=x;i;i-=i&-i) rx=rs_[rx];
                }
                else{
                    r=mid;k-=c_rs;
                    for(int i=y;i;i-=i&-i) ry=ls_[ry];
                    for(int i=x;i;i-=i&-i) rx=ls_[rx];
                }
            }
            cout<<l<<endl;
        }
    }
}

lazy数组做法（权值线段树做外壳）：

#include<bits/stdc++.h>
#define mid (l+r)/2
using namespace std;

typedef long long ll;
ll n,m,k,x,y,z,q,W,T,N,S,op,cnt,tmp,dst,cas,sum,val,tot,idx,test;

const int NN=50100;
int rot[NN*4];
int ls_[NN*320],rs_[NN*320],c[NN*320],lazy[NN*320];

void push_down(int ,int ,int );
void update(int ,int ,int ,int ,int& );
ll get_sum(int ,int ,int ,int ,int );
void update(int ,int ,int ,int ,int ,int );
ll get_kth(int ,int ,int ,int ,int ,int );

inline int read() {
    int x=0,f=1;char ch=getchar();
    for(;!isdigit(ch);ch=getchar()) if(ch=='-') f=-1;
    for(;isdigit(ch);x=(x<<3)+(x<<1)+(ch^48),ch=getchar());
    return x*f;
}

int main(){
    n=read();m=read();N=NN;
    for(int i=1;i<=m;i++){
        op=read();x=read();y=read();k=read();
        if(op==1) update(0,N,k,x,y,1);
        else printf("%d\n",get_kth(0,N,k,x,y,1));
    }
}

#define ls ls_[i]
#define rs rs_[i]
void push_down(int l,int r,int i){
    if(ls==0) ls=++tot;
    if(rs==0) rs=++tot;
    lazy[ls]+=lazy[i];
    lazy[rs]+=lazy[i];
    c[ls]+=lazy[i]*(mid-l+1);
    c[rs]+=lazy[i]*(r-mid);
    lazy[i]=0;
}

void update(int l,int r,int L,int R,int &i){
    if(i==0) i=++tot;
    if(L<=l&&r<=R){
        lazy[i]++;
        c[i]+=r-l+1;
        return ;
    }
    if(lazy[i]) push_down(l,r,i);
    if(mid>=L) update(l,mid,L,R,ls);
    if(mid<R) update(mid+1,r,L,R,rs);
    c[i]=c[ls]+c[rs];
}

ll get_sum(int l,int r,int L,int R,int i){
    if(i==0) return 0;
    if(L<=l&&r<=R) return c[i];
    if(lazy[i]) push_down(l,r,i);
    ll sum=0;
    if(mid>=L) sum+=get_sum(l,mid,L,R,ls);
    if(mid<R) sum+=get_sum(mid+1,r,L,R,rs);
    return sum;
}
#undef ls
#undef rs

#define ls i<<1
#define rs i<<1|1
void update(int l,int r,int pos,int L,int R,int i){
    update(0,N,L,R,rot[i]);
    if(l==r) return ;
    if(mid>=pos) update(l,mid,pos,L,R,ls);
    else update(mid+1,r,pos,L,R,rs);
}

ll get_kth(int l,int r,int k,int L,int R,int i){
    if(l==r) return l;
    ll r_ls=get_sum(0,N,L,R,rot[rs]);
    if(r_ls>=k) return get_kth(mid+1,r,k,L,R,rs);
    return get_kth(l,mid,k-r_ls,L,R,ls);
}