【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;}