POJ 2886 Who Gets the Most Candies? (线段树)

题意：模拟约瑟夫环。有N（1<=N<=500000）个孩子围成一圈，他们被顺时针编号为 1 到 N。每个小孩手中有一个卡片，上面有一个非 0 的数字，游戏从第 K 个小孩开始，他告诉其他小孩他卡片上的数字并离开这个圈，他卡片上的数字 A 表明了下一个离开的小孩，如果 A 是大于 0 的，则下个离开的是左手边第 A 个，如果是小于 0 的，则是右手边的第 A 个小孩。游戏将直到所有小孩都离开，在游戏中，第 p 个离开的小孩将得到 F(p) 个糖果，F(p) 是 p 的约数的个数，问谁将得到最多的糖果。输出最幸运的小孩的名字和他可以得到的糖果。
        F(p)的值直接暴力跑一下就可以了。

        因为A的值可能很大，所以会用到模运算，这就要求下标是从0开始，但是小孩的编号从1开始，所以在更新的时候，要考虑两者的转换，即从编号下标从1开始的，转换成编号下标从0开始的，再转换回去。

 

// Time 1563ms; Memory 24196K

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#define maxn 1<<20
#define mm 500010

using namespace std;

int tmp[mm],d[mm],g[mm];
char s[mm][12];
int size,n,cnt,dx;

struct line
{
	int l,r;
	int m;
}a[maxn];

void init()
{
	int i;
	for(n=1;n<size;n<<=1);
	for(i=n;i<2*n;i++)
	{
		a[i].l=a[i].r=i-n+1;
		if(i-n<size) a[i].m=1;
		else a[i].m=0;
	}
	for(i=n-1;i>0;i--)
	{
		a[i].l=a[2*i].l;
		a[i].r=a[2*i+1].r;
		a[i].m=a[2*i].m+a[2*i+1].m;
	}
}
void insert(int i,int x)
{
	if(a[i].l==a[i].r)
	{
		tmp[a[i].l-1]=++cnt;
		a[i].m=0;
		dx=a[i].l;
		return;
	}
	if(a[2*i].m>=x) insert(2*i,x);
	else insert(2*i+1,x-a[2*i].m);
	a[i].m--;
}

void calu() //建表，得到约数个数
{
	int i,j,limit;
	limit=(int)sqrt(mm*1.0);
	for(i=1;i<=limit;i++)
	{
		for(j=i+1;j*i<=mm;j++)
			g[i*j]+=2;
		g[i*i]++;
	}
}
int main()
{
	int i,k,y,p,mx;
	calu();
	while(scanf("%d%d",&size,&k)!=EOF)
	{
		init();
		cnt=0;
		memset(tmp,0,sizeof(tmp));
		for(i=0;i<size;i++)
			scanf("%s%d",s[i],&d[i]);
		p=size;
		for(;p;) 
		{
			insert(1,k);
			p--;
			if(p==0) break;
			if(d[dx-1]>0) k=((k-2+d[dx-1]%p)%p+p)%p+1; //得到要插入（删除）的是第几个
			else k=((k-1+d[dx-1]%p)%p+p)%p+1;
		}
		mx=0;
		for(i=0;i<size;i++) 
		{
			if(g[tmp[mx]]<g[tmp[i]]) mx=i;
			else if(g[tmp[mx]]==g[tmp[i]] && tmp[mx]>tmp[i]) mx=i;
		}
		printf("%s %d\n",s[mx],g[tmp[mx]]);
	}
	return 0;
}