【BZOJ3110】K大数查询(树套树)

传送门

BZOJ3110 K大数查询

I think

    题意：有N个位置，M个操作。操作有两种，每次操作如果是1 a b c的形式表示在第a个位置到第b个位置，每个位置加入一个数c。如果是2 a b c形式，表示询问从第a个位置到第b个位置，第C大的数是多少。
    算法：树套树
    思路：在外层建一颗权值线段树，维护整个序列中的权值，每一个权值线段树的节点下建一颗普通线段数（动态开点）维护该权值在区间内的分布。
    细节：对于这种树套树的题，要分清内外层节点的不同表示方法，即根节点下标对应的到底是什么。如本题中权值线段树的左右节点是用$i<<1,i<<1|1$来存的，而内层线段树是Ls[i],Rs[i]存放。
    打标记永久化也有些不习惯，好像下传会更快，然而懒得改了…
    Attention！：之前网上大多题解说数据良心没有负数，结果现在加了数据不仅有负数还要处理long long ￣へ￣）

Code

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long LL;
const int sm = 5e4+10;
const int sn = 2e7+10;
int n,q,cnt,tot,hsh;
int Rs[sn],Ls[sn];
int Rt[sm<<2];
LL Val[sn],tag[sn],B[sm];
struct Query {
    int ind,a,b;
    LL c;
}Q[sm];
int Hash(int x) {
    return lower_bound(B+1,B+cnt+1,x)-B;
}
void Insert(int &rt,int l,int r,int a,int b) {
    if(!rt) rt=++tot;
    if(a<=l&&r<=b) { Val[rt]+=(r-l+1);tag[rt]++;return;}
    int m=(l+r)>>1;
    if(a<=m) Insert(Ls[rt],l,m,a,b);
    if(b> m) Insert(Rs[rt],m+1,r,a,b);
    Val[rt]=Val[Ls[rt]]+Val[Rs[rt]]+tag[rt]*(r-l+1);
}
void Query(int &rt,int l,int r,int a,int b,LL &S,LL add) {
    if(!rt) rt=++tot;
    if(a<=l&&r<=b) { S+=Val[rt]+add*(r-l+1);return; }
    int m=(l+r)>>1;
    if(a<=m) Query(Ls[rt],l,m,a,b,S,add+tag[rt]);
    if(b> m) Query(Rs[rt],m+1,r,a,b,S,add+tag[rt]);
}
int main() {
    scanf("%d%d",&n,&q);
    for(int i=1;i<=q;++i) {
        scanf("%d%d%d%lld",&Q[i].ind,&Q[i].a,&Q[i].b,&Q[i].c);
        if(Q[i].ind==1) B[++cnt]=Q[i].c;
    }
    sort(B+1,B+cnt+1);
    cnt=unique(B+1,B+cnt+1)-B-1;
    for(int i=1;i<=q;++i) {
        int l=1,r=cnt,root=1;
        if(Q[i].ind==1) {
            hsh=Hash(Q[i].c);
            while(l<=r) {
                Insert(Rt[root],1,n,Q[i].a,Q[i].b);
                if(l==r) break;
                int m=(l+r)>>1;
                if(hsh<=m) r=m,root<<=1;
                else l=m+1,root=root<<1|1;
            }
        }
        else {  
            LL K=Q[i].c,ans;
            while(l<r) {
                Query(Rt[root<<1|1],1,n,Q[i].a,Q[i].b,ans=0,0);
                int m=(l+r)>>1;
                if(K<=ans) root=root<<1|1,l=m+1;
                else root<<=1,r=m,K-=ans;
            }
            printf("%lld\n",B[l]);
        }
    }
    return 0;
}