【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
",ask(1,nm,1,qa[i],qb[i],qc[i]));
	}
	return 0;
}