Just h-index（主席树）

Just h-index

题意：给n个数，q次询问，每次询问给出一个区间[l,r]，要你求出最大的h，使得在[l,r]这个区间内满足，有h个数的值大于等于h。

思路：二分加主席树。

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
int a[maxn],b[maxn];
int rt[maxn*20],ls[maxn*20],rs[maxn*20],sum[maxn*20];
int cnt,ql,qr,k;
void build(int& o,int l,int r)
{
	o=++cnt;
	sum[o]=0;
	if(l==r)
	return;
	int m=(l+r)/2;
	build(ls[o],l,m);
	build(rs[o],m+1,r);
}
void up(int pre,int &o,int l,int r,int pos){
	o=++cnt;
	ls[o]=ls[pre];
	rs[o]=rs[pre];
	sum[o]=sum[pre]+1;
	if(l==r)
		return;
	int m=(l+r)>>1;
	if(pos<=m)
		up(ls[pre],ls[o],l,m,pos);
	else up(rs[pre],rs[o],m+1,r,pos);
}
int qu(int pre,int o,int l,int r){
	if(l==r)
		return b[l];
	int m=(l+r)>>1;
	int cmp=sum[ls[o]]-sum[ls[pre]];
	if(cmp>=k)
		return qu(ls[pre],ls[o],l,m);
	else{
		k-=cmp;
		return qu(rs[pre],rs[o],m+1,r);
	}
}
int main(){
	int n,q;
	while(~scanf("%d%d",&n,&q)){
		cnt=0;
		for(int i=1;i<=n;i++){
			scanf("%d",&b[i]);
			a[i]=b[i];
		}
		sort(b+1,b+n+1);
		int sz=unique(b+1,b+1+n)-b-1;
		build(rt[0],1,sz);
		for(int i=1;i<=n;i++)
			a[i]=lower_bound(b+1,b+1+sz,a[i])-b;
		for(int i=1;i<=n;i++)
			up(rt[i-1],rt[i],1,sz,a[i]);
		while(q--){
			scanf("%d%d",&ql,&qr);
			int len=qr-ql+1;
			int l=1,r=len,ans=0;
			while(l<=r){
				int m=(l+r)>>1;
				k=len-m+1;
				int t=qu(rt[ql-1],rt[qr],1,sz);
				if(t>=m){
					l=m+1;
					ans=max(m,ans);
				}
				else {
					r=m-1;	
				}
			}
			printf("%d\n",ans);
		}
	}
}