ZOJ 3279 Ants

题意：先给定了蚂蚁每个等级的成员数量，然后求某只排名为x的蚂蚁的等级。显然蚂蚁的排名按照等级排列，同级的占有一段连续的区域。

做法：了解题意后，发现和POJ某题很像

#include<cstdio>
#include<cstring>
#define left l,m,x<<1
#define right m+1,r,x<<1|1
const int LMT=100003;
int have[LMT],cov[LMT<<2];
void init(void)
{
	memset(have,0,sizeof(have));
}
void update(int num,int pos,int l,int r,int x)
{
	if(l==r)
	{
		cov[x]=num;
		return;
	}
	int m=(l+r)>>1;
	if(pos<=m)update(num,pos,left);
	if(pos>m)update(num,pos,right);
	cov[x]=cov[x<<1]+cov[x<<1|1];
}
int query(int pos,int l,int r,int x)
{
	if(l==r)return l;
	int m=(l+r)>>1;
	if(cov[x<<1]>=pos)return query(pos,left);
	else
	{
		pos-=cov[x<<1];
		return query(pos,right);
	}
}
int main(void)
{
	int i,n,m,a,b,x;
	char ord;
	while(~scanf("%d",&n))
	{
		init();
		for(i=0;i<n;i++)
		{
			scanf("%d",&have[i]);
			update(have[i],i,0,n-1,1);
		}
		scanf("%d",&m);
		while(m--)
		{
			ord=getchar();
			while(ord!='p'&&ord!='q')ord=getchar();
			if(ord=='p')
			{
				scanf("%d%d",&a,&b);
				a--;
				update(b,a,0,n-1,1);
			}
			else
			{
				scanf("%d",&x);
				printf("%d\n",query(x,0,n-1,1)+1);
			}
		}
	}
			return 0;
}