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操作中c<=Maxlongint

分析：一开始的想法是区间线段树套权值线段树，码了一点后发现lazy标记没法下传，就只好强行改为权值线段树套区间线段树。
调了好久才发现是lazy标记下传时打错了一点点……
然后就T了。
于是就跑去看黄学长的程序，发现黄学长的查询操作写的比我要优美，也就是每次二分答案后权值线段树的区间同时跟着两个指针走，那么就不用每次查询的时候多加一个log啦。
顺便说一句，黄学长之前AC的程序已经无法AC了，原因貌似是没开long long……

代码：

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define N 50005
#define M 25000005
#define ll long long
using namespace std;

int n,m,tot,root[N*4];
struct tree{int l,r,lazy;ll s;}t[M];

inline int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f-=1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

inline void pushdown(int d,int l,int r)
{
    if (l==r||!t[d].lazy) return;
    int mid=(l+r)/2;
    if (!t[d].l) t[d].l=++tot;
    if (!t[d].r) t[d].r=++tot;
    t[t[d].l].s+=t[d].lazy*(mid-l+1);
    t[t[d].l].lazy+=t[d].lazy;
    t[t[d].r].s+=t[d].lazy*(r-mid);
    t[t[d].r].lazy+=t[d].lazy;
    t[d].lazy=0;
}

inline void updata(int d)
{
    t[d].s=t[t[d].l].s+t[t[d].r].s;
}

inline void ins2(int &d,int l,int r,int x,int y)
{
    if (!d) d=++tot;
    pushdown(d,l,r);
    if (l==x&&r==y)
    {
        t[d].s+=y-x+1;
        t[d].lazy=1;
        return;
    }
    int mid=(l+r)/2;
    if (y<=mid) ins2(t[d].l,l,mid,x,y);
    else if (x>mid) ins2(t[d].r,mid+1,r,x,y);
    else
    {
        ins2(t[d].l,l,mid,x,mid);
        ins2(t[d].r,mid+1,r,mid+1,y);
    }
    updata(d);
}

inline ll solve(int d,int l,int r,int x,int y)
{
    if (!d) return 0;
    pushdown(d,l,r);
    if (l==x&&r==y) return t[d].s;
    int mid=(l+r)/2;
    if (y<=mid) return solve(t[d].l,l,mid,x,y);
    else if (x>mid) return solve(t[d].r,mid+1,r,x,y);
    else return solve(t[d].l,l,mid,x,mid)+solve(t[d].r,mid+1,r,mid+1,y);
}

inline void ins1(int d,int l,int r,int x,int y,int z)
{
    ins2(root[d],1,n,x,y);
    if (l==r) return;
    int mid=(l+r)/2;
    if (z<=mid) ins1(d*2,l,mid,x,y,z);
    else ins1(d*2+1,mid+1,r,x,y,z);
}

inline int query(int x,int y,int z)
{
    int l=1,r=n;
    if (!z) return n;
    int k=1;
    ll s=0;
    while (l<r)
    {
        int mid=(l+r)/2;
        ll w=solve(root[k*2+1],1,n,x,y);
        if (s+w+1<=z)
        {
            r=mid;s+=w;
            k=k*2;
        }else
        {
            l=mid+1;k=k*2+1;
        }
    }
    return r;
}

int main()
{
    freopen("sequence9.in","r",stdin);
    freopen("test.out","w",stdout);
    n=read();m=read();
    for (int i=1;i<=m;i++)
    {
        int w=read(),x=read(),y=read(),z=read();
        if (w==1) ins1(1,1,n,x,y,z);
        else printf("%d\n",query(x,y,z));
    }
    return 0;
}