【BZOJ3110】[Zjoi2013]K大数查询 树套树

【BZOJ3110】[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

题解：考前填坑++rp。本题直接权值线段树+区间线段树即可！

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define lson x<<1
#define rson x<<1|1
using namespace std;
const int maxn=50010;
typedef long long ll;
int n,m,nn,nm,tot;
int op[maxn],qa[maxn],qb[maxn],qc[maxn];
int rt[maxn<<2],ref[maxn];
struct sag
{
int ls,rs;
ll siz,tag;
}s[maxn*400];
struct NUM
{
int org,val;
}num[maxn];
int rd()
{
int ret=0,f=1;	char gc=getchar();
while(gc<'0'||gc>'9')	{if(gc=='-')f=-f;	gc=getchar();}
while(gc>='0'&&gc<='9')	ret=ret*10+gc-'0',gc=getchar();
return ret*f;
}
void pushdown(int l,int r,int x)
{
if(s[x].tag)
{
int mid=l+r>>1;
if(!s[x].ls)	s[x].ls=++tot;
s[s[x].ls].siz+=(mid-l+1)*s[x].tag,s[s[x].ls].tag+=s[x].tag;
if(!s[x].rs)	s[x].rs=++tot;
s[s[x].rs].siz+=(r-mid)*s[x].tag,s[s[x].rs].tag+=s[x].tag;
s[x].tag=0;
}
}
void updata(int l,int r,int &x,int a,int b)
{
if(!x)	x=++tot;
if(a<=l&&r<=b)
{
s[x].tag++,s[x].siz+=r-l+1;
return ;
}
pushdown(l,r,x);
int mid=l+r>>1;
if(a<=mid)	updata(l,mid,s[x].ls,a,b);
if(b>mid)	updata(mid+1,r,s[x].rs,a,b);
s[x].siz=s[s[x].ls].siz+s[s[x].rs].siz;
}
ll query(int l,int r,int &x,int a,int b)
{
if(!x)	x=++tot;
if(a<=l&&r<=b)	return s[x].siz;
pushdown(l,r,x);
int mid=l+r>>1;
if(b<=mid)	return query(l,mid,s[x].ls,a,b);
if(a>mid)	return query(mid+1,r,s[x].rs,a,b);
return query(l,mid,s[x].ls,a,b)+query(mid+1,r,s[x].rs,a,b);
}
void insert(int l,int r,int x,int a,int b,int c)
{
updata(1,n,rt[x],a,b);
if(l==r)	return ;
int mid=l+r>>1;
if(c<=mid)	insert(l,mid,lson,a,b,c);
else	insert(mid+1,r,rson,a,b,c);
}
int ask(int l,int r,int x,int a,int b,ll c)
{
if(l==r)	return ref[l];
ll tmp=query(1,n,rt[rson],a,b);
int mid=l+r>>1;
if(tmp<c)	return ask(l,mid,lson,a,b,c-tmp);
return ask(mid+1,r,rson,a,b,c);
}
bool cmp(NUM a,NUM b)
{
return a.val<b.val;
}
int main()
{
n=rd(),m=rd();
int i;
for(i=1;i<=m;i++)
{
op[i]=rd(),qa[i]=rd(),qb[i]=rd(),qc[i]=rd();
if(op[i]==1)	num[++nn].val=qc[i],num[nn].org=i;
}
sort(num+1,num+nn+1,cmp);
num[0].val=-1<<30;
for(i=1;i<=nn;i++)
{
if(num[i].val>num[i-1].val)	ref[++nm]=num[i].val;
qc[num[i].org]=nm;
}
for(i=1;i<=m;i++)
{
if(op[i]==1)	insert(1,nm,1,qa[i],qb[i],qc[i]);
else	printf("%d\n",ask(1,nm,1,qa[i],qb[i],qc[i]));
}
return 0;
}