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;
}