bzoj 3110 [Zjoi2013]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

輸出每個詢問的結果

Sample Input

2 5
1 1 2 1
1 1 2 2
2 1 1 2
2 1 1 1
2 1 2 3

Sample Output

1
2
1

HINT

【樣例說明】

第一個操作 後位置 1 的數只有 1 ， 位置 2 的數也只有 1 。 第二個操作 後位置 1 的數有 1 、 2 ，位置 2 的數也有 1 、 2 。 第三次詢問 位置 1 到位置 1 第 2 大的數 是 1 。 第四次詢問 位置 1 到位置 1 第 1 大的數是 2 。 第五次詢問 位置 1 到位置 2 第 3 大的數是 1 。‍

N,M<=50000,N,M<=50000

a<=b<=N

1操作中abs(c)<=N

2操作中abs(c)<=Maxlongint

Key To Problem

據說是線段樹套線段樹，然而我不服啊，於是就寫的線段樹套平衡樹，寫着寫着之後就JJ了，所以還是老老實實的寫線段樹吧。。
第一位記錄權值，第二位記錄區間，雖然常數寫的有點大，但畢竟是20s的bz題，不虛不虛。
對於權值在[l,r]的數，建立一棵普通線段樹，求解時可二分答案。

Code

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define N 50010
#define ls tr[rt].l
#define rs tr[rt].r
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
using namespace std;
struct node
{
    int l,r,lasy,sum;
};
node tr[(N<<2)*100];
int n,m,size;
int root[N<<2];

void PushDown(int l,int r,int rt)
{
    if(!tr[rt].lasy||l==r)return ;
    if(!tr[rt].l)tr[rt].l=++size;
    if(!tr[rt].r)tr[rt].r=++size;
    int mid=(l+r)>>1;
    tr[ls].lasy+=tr[rt].lasy;
    tr[rs].lasy+=tr[rt].lasy;
    tr[ls].sum+=tr[rt].lasy*(mid-l+1);
    tr[rs].sum+=tr[rt].lasy*(r-mid);
    tr[rt].lasy=0;
}

void PushUp(int rt)
{
    tr[rt].sum=tr[ls].sum+tr[rs].sum;
}

void insert(int l,int r,int &rt,int L,int R)
{
    if(!rt)rt=++size;
    PushDown(l,r,rt);
    if(l==L&&r==R)
    {
        tr[rt].sum+=r-l+1;
        tr[rt].lasy++;
        return ;
    }
    int mid=(l+r)>>1;
    if(mid>=R)insert(l,mid,tr[rt].l,L,R);
    else if(mid<L)insert(mid+1,r,tr[rt].r,L,R);
    else{
        insert(l,mid,tr[rt].l,L,mid);
        insert(mid+1,r,tr[rt].r,mid+1,R);
    }
    PushUp(rt);
}

void build(int l,int r,int rt,int L,int R,int c)
{
    insert(1,n,root[rt],L,R);
    if(l==r)return ;
    int mid=(l+r)>>1;
    if(mid>=c)build(lson,L,R,c);
    else build(rson,L,R,c);
}

int ask(int l,int r,int rt,int L,int R)
{
    if(!rt)return 0;
    PushDown(l,r,rt);
    if(l==L&&r==R)return tr[rt].sum;
    int mid=(l+r)>>1;
    if(mid>=R)return ask(l,mid,tr[rt].l,L,R);
    else if(mid<L)return ask(mid+1,r,tr[rt].r,L,R);
    else return ask(l,mid,tr[rt].l,L,mid)+ask(mid+1,r,tr[rt].r,mid+1,R);
}

int query(int l,int r,int rt,int L,int R,int c)
{
    if(l==r)return l;
    int mid=(l+r)>>1;
    int t=ask(1,n,root[rt<<1|1],L,R);
    if(t+1<=c){
        c-=t;
        return query(lson,L,R,c);
    }else return query(rson,L,R,c);
}

int main()
{
    cin>>n>>m;
    for(int i=1;i<=m;i++)
    {
        int opt,x,y,z;
        scanf("%d%d%d%d",&opt,&x,&y,&z);
        switch(opt)
        {
            case 1:build(1,n,1,x,y,z);break;
            case 2:printf("%d\n",query(1,n,1,x,y,z));break;
        }
    }
    return 0;
}