BZOJ 3110: [Zjoi2013]K大数查询(整体二分)

3110: [Zjoi2013]K大数查询

Time Limit: 20 Sec  Memory Limit: 512 MB
Submit: 11673  Solved: 3512
[Submit][Status][Discuss]

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

输出每个询问的结果

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=1e5+10;
int n,m,ans[maxn],now1,now2;
struct node
{
    int op,a,b,id;
    ll cnt,c;
}t[maxn],q1[maxn],q2[maxn];
ll tree[maxn<<2],add[maxn<<2];
void push_down(int l,int r,int rt)
{
    if(add[rt]!=0)
    {
        int mid=(l+r)/2;
        tree[rt<<1]+=(mid-l+1)*add[rt];
        tree[rt<<1|1]+=(r-mid)*add[rt];
        add[rt<<1]+=add[rt];
        add[rt<<1|1]+=add[rt];
        add[rt]=0;
    }
}
void update(int L,int R,int C,int l,int r,int rt)
{
    if(l>=L&&r<=R)
    {
        tree[rt]+=(r-l+1)*C;
        add[rt]+=C;
        return;
    }
    push_down(l,r,rt);
    int mid=(l+r)/2;
    if(L<=mid) update(L,R,C,l,mid,rt<<1);
    if(R>mid) update(L,R,C,mid+1,r,rt<<1|1);
    tree[rt]=tree[rt<<1]+tree[rt<<1|1];
}
ll query(int L,int R,int l,int r,int rt)
{
    if(l>=L&&r<=R) return tree[rt];
    push_down(l,r,rt);
    ll ans=0;
    int mid=(l+r)/2;
    if(L<=mid) ans+=query(L,R,l,mid,rt<<1);
    if(R>mid) ans+=query(L,R,mid+1,r,rt<<1|1);
    return ans;
}
void cal(int L,int R,int mid,int r)
{
    for(int i=L;i<=R;i++)
    {
        if(t[i].op==2) t[i].cnt=query(t[i].a,t[i].b,1,n,1);
        else if(t[i].op==1&&t[i].c>=mid) update(t[i].a,t[i].b,1,1,n,1);
    }
    for(int i=L;i<=R;i++)
        if(t[i].op==1&&t[i].c>=mid) update(t[i].a,t[i].b,-1,1,n,1);
    now1=now2=0;
    for(int i=L;i<=R;i++)
    {
        if(t[i].op==1)
        {
            if(t[i].c<mid) q1[++now1]=t[i];
            else q2[++now2]=t[i];
        }
        else
        {
            if(t[i].cnt>=t[i].c) q2[++now2]=t[i];
            else t[i].c-=t[i].cnt,q1[++now1]=t[i];
        }
    }
    int r1=L;
    for(int i=1;i<=now1;i++) t[r1++]=q1[i];
    for(int i=1;i<=now2;i++) t[r1++]=q2[i];
}
void work(int L,int R,int l,int r)
{
    if(l==r)
    {
        for(int i=L;i<=R;i++)
            if(t[i].op==2) ans[t[i].id]=l;
        return;
    }
    int mid=l+(r-l)/2;
    cal(L,R,mid+1,r);
    int tmp=now1;
    if(tmp) work(L,L+tmp-1,l,mid);
    if(L+tmp<=R) work(L+tmp,R,mid+1,r);
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d%lld",&t[i].op,&t[i].a,&t[i].b,&t[i].c);
        t[i].cnt=0;t[i].id=i;
    }
    work(1,m,-n,n);
    for(int i=1;i<=m;i++) if(ans[i]) printf("%d\n",ans[i]);
    return 0;
}