【ZJOI2013】bzoj3110 K大数查询【解法一】

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

输出每个询问的结果

整体二分做法见【这里】。
权值线段树套区间线段树，区间线段树上维护元素个数。
对于修改操作，在权值线段树上从顶到下朝着要修改的权值走一遍，对经过的区间线段树进行区间修改。
对于查询操作，在权值线段树上二分，每一次进入区间线段树中查询个数。
注意$n*q$会爆int，需要unsigned int。

#include<cstdio>
#include<algorithm>
using namespace std;
int rd()
{
    int x=0,f=1;
    char c=getchar();
    while (c<'0'||c>'9')
    {
        if (c=='-') f=-1;
        c=getchar();
    }
    while (c>='0'&&c<='9')
    {
        x=x*10+c-'0';
        c=getchar();
    }
    return x*f;
}
unsigned int rdu()
{
    unsigned int x=0;
    char c=getchar();
    while (c<'0'||c>'9') c=getchar();
    while (c>='0'&&c<='9')
    {
        x=x*10+c-'0';
        c=getchar();
    }
    return x;
}
unsigned int sum[25000010],tag[25000010],K[50010];
int root[25000010],son[25000010][2],
opt[50010],L[50010],R[50010],tem[50010],ord[50010],
n,m,q,clo;
void down(int p,int L,int R)
{
    if (tag[p])
    {
        sum[p]+=tag[p]*(R-L+1);
        if (L==R)
        {
            tag[p]=0;
            return;
        }
        if (!son[p][0]) son[p][0]=++clo;
        tag[son[p][0]]+=tag[p];
        if (!son[p][1]) son[p][1]=++clo;
        tag[son[p][1]]+=tag[p];
        tag[p]=0;
    }
}
void up(int p,int L,int R)
{
    sum[p]=0;
    int mid=L+R>>1;
    if (son[p][0])
    {
        down(son[p][0],L,mid);
        sum[p]+=sum[son[p][0]];
    }
    if (son[p][1])
    {
        down(son[p][1],mid+1,R);
        sum[p]+=sum[son[p][1]];
    }
}
void modify(int &p,int L,int R,int l,int r)
{
    if (!p) p=++clo;
    down(p,L,R);
    if (l<=L&&R<=r)
    {
        tag[p]++;
        down(p,L,R);
        return;
    }
    int mid=L+R>>1;
    if (l<=mid) modify(son[p][0],L,mid,l,r);
    if (r>mid) modify(son[p][1],mid+1,R,l,r);
    up(p,L,R);
}
void Modify(int p,int L,int R,int x,int l,int r)
{
    modify(root[p],1,n,l,r);
    if (L==R) return;
    int mid=L+R>>1;
    if (x<=mid) Modify(p<<1,L,mid,x,l,r);
    else Modify(p<<1|1,mid+1,R,x,l,r);
}
unsigned int query(int p,int L,int R,int l,int r)
{
    if (!p) return 0;
    down(p,L,R);
    if (l<=L&&R<=r) return sum[p];
    int mid=L+R>>1;
    unsigned int ret=0;
    if (l<=mid) ret+=query(son[p][0],L,mid,l,r);
    if (r>mid) ret+=query(son[p][1],mid+1,R,l,r);
    return ret;
}
int Query(int p,int L,int R,unsigned int x,int l,int r)
{
    if (L==R) return L;
    int mid=L+R>>1;
    unsigned int y=query(root[p<<1|1],1,n,l,r);
    if (y>=x) return Query(p<<1|1,mid+1,R,x,l,r);
    return Query(p<<1,L,mid,x-y,l,r);
}
int main()
{
    /*freopen("in.txt","r",stdin);
    freopen("out.txt","w",stdout);*/
    n=rd();
    q=rd();
    for (int i=1;i<=q;i++)
    {
        opt[i]=rd();
        L[i]=rd();
        R[i]=rd();
        if (opt[i]==1) tem[i]=ord[++m]=rd();
        else K[i]=rdu();
    }
    sort(ord+1,ord+m+1);
    m=unique(ord+1,ord+m+1)-ord-1;
    for (int i=1;i<=q;i++)
        if (opt[i]==1)
            K[i]=lower_bound(ord+1,ord+m+1,tem[i])-ord;
    for (int i=1;i<=q;i++)
        if (opt[i]==1)
            Modify(1,1,m,K[i],L[i],R[i]);
        else
            printf("%d\n",ord[Query(1,1,m,K[i],L[i],R[i])]);
}