BZOJ3110: [Zjoi2013]K大数查询

n<=50000个vector进行m<=50000次操作：1、一区间的vector插入一个数；2、查询一区间中的vector中的所有数的第k大。

方法一：普通线段树套权值线段树！……

方法二：单个询问是可以二分的，那就整体二分吧。这次是用线段树维护，把所有数字反过来求第k小再反过来就是第k大，注意50000*50000>int。

#include<string.h>
#include<stdlib.h>
#include<stdio.h>
//#include<assert.h>
#include<algorithm>
//#include<iostream>
using namespace std;

int n,m;
#define maxn 100011
#define LL long long
struct Ope
{
    int x,y; LL z; int id,type;
    //type=1:xiugai,[x,y] ba zhi z jia 1
    //type=0:chaxun,[x,y] cha di z da,xunwen bian hao id
}q[maxn],al[maxn],ar[maxn];int len=0;
int ans[maxn];

struct SMT
{
    struct Node
    {
        int ls,rs;
        int add; LL sum;
    }a[maxn<<1];
    int size,root;
    void clear() {size=root=0;}
    void up(int x)
    {
        const int &p=a[x].ls,&q=a[x].rs;
        a[x].sum=a[p].sum+a[q].sum;
    }
    void addsingle(int x,int L,int R,int v)
    {
        a[x].sum+=(R-L+1)*1ll*v;
        a[x].add+=v;
    }
    void down(int x,int L,int R)
    {
        int mid=(L+R)>>1,&p=a[x].ls,&q=a[x].rs;
        if (!p) {p=++size; a[p].ls=a[p].rs=a[p].sum=a[p].add=0;}
        if (!q) {q=++size; a[q].ls=a[q].rs=a[q].sum=a[q].add=0;}
        if (a[x].add) addsingle(p,L,mid,a[x].add),addsingle(q,mid+1,R,a[x].add),a[x].add=0;
    }
    int ql,qr,v;
    void Add(int &x,int L,int R)
    {
        if (!x) {x=++size; a[x].add=a[x].sum=a[x].ls=a[x].rs=0;}
        if (ql<=L && R<=qr) {addsingle(x,L,R,v); return;}
        down(x,L,R);
        const int mid=(L+R)>>1;
        if (ql<=mid) Add(a[x].ls,L,mid+0);
        if (qr> mid) Add(a[x].rs,mid+1,R);
        up(x);
    }
    void add(int L,int R,int v)
    {
        ql=L; qr=R; this->v=v;
        Add(root,1,n);
    }
    LL Query(int &x,int L,int R)
    {
        if (!x) {x=++size; a[x].add=a[x].sum=a[x].ls=a[x].rs=0;}
        if (ql<=L && R<=qr) return a[x].sum;
        down(x,L,R);
        const int mid=(L+R)>>1; LL ans=0;
        if (ql<=mid) ans+=Query(a[x].ls,L,mid+0);
        if (qr> mid) ans+=Query(a[x].rs,mid+1,R);
        return ans;
    }
    LL query(int L,int R)
    {
        ql=L; qr=R;
        return Query(root,1,n);
    }
}t;

void solve(int L,int R,int ql,int qr)
{
    if (L>R || ql>qr) return;
    if (L==R)
    {
        for (int i=ql;i<=qr;i++) if (q[i].type==0) ans[q[i].id]=L;
        return;
    }
    t.clear();
    int mid=(L+R)>>1,lal=0,lar=0;
    for (int i=ql;i<=qr;i++)
    {
        if (q[i].type==1)
        {
            if (q[i].z<=mid) t.add(q[i].x,q[i].y,1),al[++lal]=q[i];
            else ar[++lar]=q[i];
        }
        else
        {
            LL tmp=t.query(q[i].x,q[i].y);
            if (tmp>=q[i].z) al[++lal]=q[i];
            else q[i].z-=tmp,ar[++lar]=q[i];
        }
    }
    for (int i=1,j=ql;i<=lal;i++,j++) q[j]=al[i];
    for (int i=1,j=ql+lal;i<=lar;i++,j++) q[j]=ar[i];
    solve(L,mid,ql,ql+lal-1);
    solve(mid+1,R,ql+lal,qr);
}

int main()
{
    scanf("%d%d",&n,&m);
    int cntq=0;
    for (int i=1,op;i<=m;i++)
    {
        scanf("%d",&op);
        if (op==1) len++,scanf("%d%d%lld",&q[len].x,&q[len].y,&q[len].z),q[len].z*=-1,q[len].type=1;
        else len++,scanf("%d%d%lld",&q[len].x,&q[len].y,&q[len].z),q[len].id=++cntq,q[len].type=0;
    }
    solve(-n,n,1,len);
    for (int i=1;i<=cntq;i++) printf("%d
",-ans[i]);
    return 0;
}