【BZOJ3110】【ZJOI2013】K大数查询 题解

题面：传送门

---

    一看就是一道毒瘤数据结构题，因为既有修改又有第$k$大查询，所以我们可以考虑树套树，但是线段树套平衡树是显然不可行的，每次修改的时候懒标记可能会使空间复杂度变为$\Theta(n^2)$。考虑线段树套线段树，那这题一定是一棵区间线段树和一棵权值线段树，那谁在里谁在外呢？由于每次都是将一个区间全部加入一个相同的数，所以权值线段树必须在外。

    这道题因为要动态开点还是比较难写的，我$wa$了好几发，调了半天才调对。由于是两棵线段树，所以修改的复杂度是$\Theta(mlog_2^2n)$，查询二分答案的过程可以与权值线段树合并，所以复杂度也是$\Theta(mlog_2^2n)$。最终复杂度：$\Theta(mlog_2^2n)$。（据说整体二分更快，但我不会QAQ）

    全部代码：

#pragma GCC optimize(2)
#include<iostream>
#include<cstdio>
#include<cstdlib>
using namespace std;
int n,m;
namespace sgt2
{
    long long sum[20000010]; 
    int chg[20000010],son[20000010][2],cnt;
    int nw(int &x)
    {
        if(!x)return x=++cnt;
        return x;
    }
    void update(int a,int b,int k,int l,int r)
    {
        if(a==l && b==r)
        {
            chg[k]++;
            sum[k]+=r+1-l;
        }
        else
        {
            int mid=(l+r)>>1;
            if(b<=mid)update(a,b,nw(son[k][0]),l,mid);
            else if(a>mid)update(a,b,nw(son[k][1]),mid+1,r);
            else
            {
                update(a,mid,nw(son[k][0]),l,mid);
                update(mid+1,b,nw(son[k][1]),mid+1,r);
            }
            sum[k]+=b+1-a;
        }
    }
    long long query(int a,int b,int k,int l,int r)
    {
        if(a==l && b==r)return sum[k];
        else
        {
            long long res=0;
            int mid=(l+r)>>1;
            if(b<=mid)res=query(a,b,nw(son[k][0]),l,mid);
            else if(a>mid)res=query(a,b,nw(son[k][1]),mid+1,r);
            else
            {
                res=query(a,mid,nw(son[k][0]),l,mid)+query(mid+1,b,nw(son[k][1]),mid+1,r);
            }
            res+=1LL*(b+1-a)*chg[k];
            return res;
        }
    }
}
namespace sgt1
{
    int root[1000010],cnt,rt,son[1000010][2];
    void build(int &x,int l,int r)
    {
        x=++cnt;
        if(l==r)return;
        int mid=(l+r)>>1;
        build(son[x][0],l,mid);
        build(son[x][1],mid+1,r);
    }
    void update(int x,int l,int r,int v,int L,int R)
    {
        sgt2::update(L,R,sgt2::nw(root[x]),1,n);
        if(l==r)return;
        int mid=(l+r)>>1;
        if(v<=mid)update(son[x][0],l,mid,v,L,R);
        else update(son[x][1],mid+1,r,v,L,R);
    }
    int query(int x,int l,int r,long long v,int L,int R)
    {
        if(l==r)return l;
        long long res=sgt2::query(L,R,sgt2::nw(root[son[x][1]]),1,n);
        int mid=(l+r)>>1;
        if(res>=v)return query(son[x][1],mid+1,r,v,L,R);
        else return query(son[x][0],l,mid,v-res,L,R);
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    sgt1::build(sgt1::rt,1,n);
    while(m--)
    {
        int op;
        scanf("%d",&op);
        if(op==1)
        {
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            sgt1::update(sgt1::rt,1,n,c,a,b);
        }
        else
        {
            int a,b;
            long long c;
            scanf("%d%d%lld",&a,&b,&c);
            printf("%d\n",sgt1::query(sgt1::rt,1,n,c,a,b));
        }
    }
    return 0;
}