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;
}