BZOJ 3110 K大数查询 树套树

题意：链接

方法：树套树（线段树套线段树）

题解：这题好神啊- -自己在做的时候一顿yy也没yy出用两个线段树来搞，其实我想的是类似二逼那道题那样，用线段树维护总区间，treap维护每个节点，不过这样的话，尼玛修改就是暴力有没有？而且查询的时候也是暴力啊有没有？绝壁不是这么做的啊！

上网上找了找题解综合了大家的思想自己也是懂了这题是咋回事了，也是跪了。

好不扯淡了，谈正经的，两个线段树是怎么搞得。

其实第一棵线段树是区间的线段树，而第二棵是维护值域的线段树，具体解析请看代码。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define N 20000100
using namespace std ;
int lson[N],rson[N];
int sum[N],col[N];
int root[N];
int n,m,size;
int a,b,c;
//这里的sum是针对于值域为l~r,区间为L~R
//本代码的所有l,r都是代表值域，而L,R都是代表区间
void pushup(int rt)
{
    sum[rt]=sum[lson[rt]]+sum[rson[rt]];
}
void pushdown(int rt,int l,int r)
{
    if(!col[rt]||l==r)return;
    if(!lson[rt])lson[rt]=++size;//动态加点
    if(!rson[rt])rson[rt]=++size;//动态加点
    col[lson[rt]]+=col[rt];
    col[rson[rt]]+=col[rt];
    int mid=(l+r)>>1;
    sum[lson[rt]]+=(mid-l+1)*col[rt];
    sum[rson[rt]]+=(r-mid)*col[rt];
    col[rt]=0;
}
void update(int &rt,int l,int r,int L,int R)
{
    if(!rt)rt=++size;
    pushdown(rt,l,r);
    if(L==l&&r==R)
    {
        sum[rt]+=r-l+1;//每一次添加相当于对于区间L~R每个里面都加入了一个值域在l~r内的数，所以个数为这个。
        col[rt]++;//标记
        return;
    }
    int mid=(l+r)>>1;
    if(R<=mid)update(lson[rt],l,mid,L,R);
    else if(L>mid)update(rson[rt],mid+1,r,L,R);
    else
    {
        update(lson[rt],l,mid,L,mid);
        update(rson[rt],mid+1,r,mid+1,R);
    }
    pushup(rt);
}
int query(int rt,int l,int r,int L,int R)//查询并没有什么可说的
{
    if(!rt)return 0;
    pushdown(rt,l,r);
    if(L==l&&r==R)return sum[rt];
    int mid=(l+r)>>1;
    if(R<=mid)return query(lson[rt],l,mid,L,R);
    else if(L>mid)return query(rson[rt],mid+1,r,L,R);
    else return query(lson[rt],l,mid,L,mid)+query(rson[rt],mid+1,r,mid+1,R);
}
int work(int L,int R)//这是对应查询的
{
    int l=1,r=n,rt=1;//对于值域进行二分。最后分到的一定就是K大数。
    while(l!=r)
    {
        int mid=(l+r)>>1;
        int ans=query(root[rt<<1|1],1,n,L,R);
        if(ans<c){c-=ans,r=mid,rt<<=1;}//如果值域右子数的数量不足c，则再到左子树里查，此时c-=ans；
        else{l=mid+1,rt=rt<<1|1;}
    }
    return r;
}
void insert(int L,int R)
{
    int l=1,r=n,rt=1;
    while(l!=r)
    {
        update(root[rt],1,n,L,R);
        int mid=(l+r)>>1;
        if(c<=mid){r=mid,rt<<=1;}
        else{l=mid+1,rt=rt<<1|1;}
    }
    update(root[rt],1,n,L,R);//不要忘- -单点的区间也要更新。
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
    {
        int jd;
        scanf("%d%d%d%d",&jd,&a,&b,&c);
        if(jd==1)
        {
            insert(a,b);
        }else printf("%d\n",work(a,b));
    }
}