HDU - 6278 Just $h$-index主席树+二分

HDU - 6278 Just $h$-index
【题目描述】

【题目分析】
题目要求在区间$[l,r]$内大于h的数不少于h个，对于这种最大化问题，我们应该想到二分。
最小情况显然是1.最大情况显然是$r-l+1$，对于一个$h$，我们如何判断能否满足条件呢？
我们可以用主席树方便的求出区间第h大，如果区间第h大大于等于h，那么就能满足条件
比较伤心的是我虽然一开始就想到了正确的思路但是区间第k大却写错了。因为之前一直写的是将所有数据都放在结构体中，所以一个赋值就讲所有数据复制了，但是这次我想写简单一点，就将计数的数组放在了外面，然后在创建新节点的时候却忘记了复制，导致疯狂wa，还一直找不到为什么，啊啊啊，最终还是调试找到了
【AC代码】（没有借鉴其他人的，嘿嘿）

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#include<vector>
#include<set>
#include<climits>
#include<string>
#include<cmath>
#include<cstdlib>

using namespace std;

const int MAXN=100005;
int a[MAXN];
int n,m;
struct node
{
	int ls,rs;
}tree[MAXN*40];
int root[MAXN*40];
int cnt[MAXN*40];
int top;

void Insert(int &k,int x,int l,int r)
{
	tree[++top]=tree[k]; 
	cnt[top]=cnt[k];
	cnt[top]++; k=top;
	//printf("l=%d r=%d x=%d cnt[%d]=%d\n",l,r,x,k,cnt[k]);
	if(l==r) return;
	int mid=(l+r)>>1;
	if(x<=mid) Insert(tree[k].ls,x,l,mid);
	else Insert(tree[k].rs,x,mid+1,r);
}

int query(int i,int j,int k,int l,int r)
{
	if(l==r) return l;
	int tmp=cnt[tree[j].rs]-cnt[tree[i].rs];
	int mid=(l+r)>>1;
	if(k<=tmp) return query(tree[i].rs,tree[j].rs,k,mid+1,r);
	else return query(tree[i].ls,tree[j].ls,k-tmp,l,mid);
}

int main()
{
	int u,v,l,r,mid,ans;
	while(~scanf("%d%d",&n,&m))
	{
		memset(cnt,0,sizeof(cnt));
		memset(root,0,sizeof(root));
		memset(tree,0,sizeof(tree));
		top=0;
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
			root[i]=root[i-1];
			Insert(root[i],a[i],1,n);
		}
		/*
		for(int i=1;i<=n;i++)
		{
			printf("%d ",query(root[0],root[n],i,1,n));
		}
		*/
		
		for(int i=0;i<m;i++)
		{
			scanf("%d%d",&u,&v);
			l=1; r=v-u+1;
			while(l<=r)
			{
				mid=(l+r)>>1;
				if(query(root[u-1],root[v],mid,1,n)>=mid)
				{
					ans=mid;
					l=mid+1;
				}
				else
				{
					r=mid-1;
				}
			}
			printf("%d\n",ans);
		}
	}
	return 0;
}