POJ-2886 Who Gets the Most Candies? 线段树|树状数组

          题目链接：http://poj.org/problem?id=2886

       题意：一群孩子围成一个圈（顺时针），每个孩子手上有一个号码，+A代表他左边的第A个孩子，-A代表他右边的第A个孩子。从第K个孩子开始，依次去掉第K+Ak个孩子，如果那个孩子是第p个去除的，那么他可以得到p所有约数的个数个糖果，求哪个孩子能得到最多的糖果数。

       先用筛选法求出约数个数，然后用线段树的统计操作，统计区间里有多少个孩子被去除，直到最大的那个为止。

   My code:

//STATUS:C++_AC_1641MS_17792KB  
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<vector>
#include<queue>
#include<math.h>
#include<map>
#include<set>
using namespace std;
#define LL __int64
#define pii pair<int,int>
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
const int MAX=500010,INF=200000000;
const double esp=1e-6;
void update(int l,int r,int rt);
void getw();
struct People{
	char name[12];
	int v;
}a[MAX];
int ret[MAX<<2],p[MAX];
int n,k,w,tot,cou,ans;
int main()
{
//	freopen("in.txt","r",stdin);
	int i,j,max_sqrtn,maxw,max;
	max_sqrtn=MAX/2;
	
	for(i=2;i<=max_sqrtn;i++)
		for(j=i+i;j<MAX;j+=i)
			p[j]++;
	p[1]=-1;
	while(~scanf("%d%d",&n,&k))
	{
		memset(ret,0,sizeof(ret));
		for(i=2,maxw=1,max=-1;i<=n;i++)
			if(p[i]>max){
				max=p[i];
				maxw=i;
			}
		for(i=1;i<=n;i++)
			scanf("%s%d",a[i].name,&a[i].v);

		for(i=1,w=k,tot=n;i<=n;i++){
			cou=w;
			update(1,n,1);
			if(i==maxw)break;
			tot--;
			getw();
		}

		printf("%s %d\n",a[ans].name,p[maxw]+2);
	}
	return 0;
}

void getw()
{
	int lsum;
	lsum=w-1;
	if(a[ans].v>0)
		w=(lsum+a[ans].v)%tot;	
	else 
		w=((lsum+a[ans].v)%tot+tot)%tot+1;
	if(!w)w=tot;
}

void update(int l,int r,int rt)
{
	if(l==r){
		ans=l;
		ret[rt]=1;
		return;
	}
	int mid=(l+r)>>1,t=mid-l+1-ret[rt<<1];
	if(t>=cou)
		update(lson);
	else {
		cou-=t;
		update(rson);
	}
	ret[rt]=ret[rt<<1]+ret[rt<<1|1];
}