【BZOJ3110】【ZJOI2013】K大数查询

3110: [Zjoi2013]K大数查询

Time Limit: 20 Sec Memory Limit: 512 MB
Description

有N个位置，M个操作。操作有两种，每次操作如果是1 a b c的形式表示在第a个位置到第b个位置，每个位置加入一个数c
如果是2 a b c形式，表示询问从第a个位置到第b个位置，第C大的数是多少。

Input

第一行N，M
接下来M行，每行形如1 a b c或2 a b c

Output

输出每个询问的结果

Sample Input

2 5

1 1 2 1

1 1 2 2

2 1 1 2

2 1 1 1

2 1 2 3

Sample Output

1

2

1
HINT

【样例说明】

第一个操作 后位置 1 的数只有 1 ， 位置 2 的数也只有 1 。 第二个操作 后位置 1

的数有 1 、 2 ，位置 2 的数也有 1 、 2 。 第三次询问 位置 1 到位置 1 第 2 大的数 是

1 。 第四次询问 位置 1 到位置 1 第 1 大的数是 2 。 第五次询问 位置 1 到位置 2 第 3

大的数是 1 。‍

N,M<=50000,N,M<=50000

a<=b<=N

1操作中abs(c)<=N

2操作中abs(c)<=Maxlongint

Source

啊这道题解法貌似不少据说有三种，这里详细下线段树套线段树。
第一层线段树是权值线段树，它的每个节点对应了整个区间的线段树，并且第二层的线段树维护了第一层权值区间中所有数出现的总次数。
插入一种颜色c的时候，在第一层相当于单点修改，会递归地影响线段树上O(logn)个结点，对于影响到的每个节点，都只需要再在第二层线段树上做一次区间修改即可（之所以要将权值放在第一层，就是因为要在第二层线段树上的节点打lazy标记，而我实在是想不出来第一层的线段树该如何处理lazy的下放问题……）。为了减少常数，不妨在实现时使lazy标记永久化。
查询时，用类似于二分查找的办法（这个貌似跟主席树是类似的），假设当前已经确定了答案的区间为[l,r]，设其中点为m，第二层线段树查询的区间为[ql,qr]，由于是查询第k大的数，我们便查询在[ql,qr]中，所有在[m+1,r]范围内的颜色出现的总次数t，若t>=k，则说明答案在右区间；反之，说明答案在左区间，同时将k减去t。
代码真是比想象中的短多了……
BTW，本题良心数据没有负数。

#include<iostream>
#include<cstdio>
#define maxn 50000
#define maxm 20000000
#define mid (l+r>>1)
using namespace std;
int lc[maxm+10],rc[maxm+10],sum[maxm+10],val[maxm+10],root[(maxn<<4)+10],ql,qr,cnt;
void update(int &o,int l,int r){
    if(!o)o=++cnt;
    if(ql<=l&&r<=qr)val[o]++,sum[o]+=r-l+1;
    else{
        int m=mid;
        if(ql<=m)update(lc[o],l,m);
        if(qr>m)update(rc[o],m+1,r);
        sum[o]=sum[lc[o]]+sum[rc[o]]+val[o]*(r-l+1);
    }
}
void query(int &o,int l,int r,int &qsum,int add){
    if(!o)o=++cnt;
    if(ql<=l&&r<=qr)qsum+=sum[o]+add*(r-l+1);
    else{
        int m=mid;
        if(ql<=m)query(lc[o],l,m,qsum,add+val[o]);
        if(qr>m)query(rc[o],m+1,r,qsum,add+val[o]);
    }
}
int main(){
    int n,q;
    scanf("%d%d",&n,&q);
    int k,a,b,c;
    while(q--){
        scanf("%d%d%d%d",&k,&ql,&qr,&c);
        int o=1,l=1,r=n;
        if(k==1){
            for(;;){
                update(root[o],1,n);
                if(l==r)break;
                int m=mid;
                if(c<=m)o=o<<1,r=m;
                else o=o<<1|1,l=m+1;
            }
        }else{
            int t;
            for(;;){
                if(l==r)break;
                int m=mid;
                query(root[o<<1|1],1,n,t=0,0);
                if(t>=c){
                    o=o<<1|1;
                    l=m+1;
                }else{
                    c-=t;
                    o=o<<1;
                    r=m;
                }
            }
            printf("%d\n",l);
        }
    }
    return 0;
}