首页 > 代码库 > 【BZOJ3110】[Zjoi2013]K大数查询 树套树

【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|1using 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;}

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