【模板】主席树模板

#include

using namespace std ;

const int maxn = 2e5+7 ;

int n,m ;
int a[maxn],b[maxn],rt[maxn<<5],ndnum ;
int lson[maxn<<5],rson[maxn<<5],sum[maxn<<5] ;

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


void build(int &t,int l,int r)
{
	t=++ndnum ;
	if(l==r) return ;
	int mid=(l+r)/2 ;
	build(lson[t],l,mid) ;
	build(rson[t],mid+1,r) ;
}

void copy(int t,int s)
{
	lson[s]=lson[t] ;
	rson[s]=rson[t] ;
	sum[s]=sum[t]+1 ;
}

int modify(int t,int l,int r,int val)
{
	int s=++ndnum ;
	copy(t,s) ;
	if(l==r)
		return s ;
	int mid=(l+r)/2 ;
	if(val<=mid) lson[s]=modify(lson[t],l,mid,val) ;
	else rson[s]=modify(rson[t],mid+1,r,val) ;
	return s ;
}

int query(int t,int s,int l,int r,int k)
{
	if(l==r)
		return l ;
	int mid=(l+r)/2 ,x=sum[lson[s]]-sum[lson[t]] ;
	if(k<=x) return query(lson[t],lson[s],l,mid,k) ;
	else return query(rson[t],rson[s],mid+1,r,k-x) ;
}
int main()
{
	read(n) ;read(m) ;
	for(int i=1;i<=n;++i)
		read(a[i]) ,b[i]=a[i] ;
	sort(b+1,b+1+n) ;
	int tot=unique(b+1,b+1+n)-b-1 ;
	build(rt[0],1,tot) ;
	for(int i=1;i<=n;++i)
	{
		int p=lower_bound(b+1,b+1+tot,a[i])-b ;
		rt[i]=modify(rt[i-1],1,tot,p) ;
	}
	while(m--)
	{
		int l,r,k ;read(l) ;read(r) ;read(k) ;
		int id=query(rt[l-1],rt[r],1,tot,k) ;
		printf("%d\n",b[id]) ;
	}
	return 0 ; 	
} 

你可能感兴趣的:(数据结构,线段树)