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