POJ 2886: Who Gets the Most Candies?

这道题拖了好久，眼泪都快掉下来了。。。

先放个偷懒版代码，

等我学了反素数再把打表那部分完善一下。

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

题目大意：

n个孩子围成一圈。

第k个孩子先退出圈子，

然后根据他手持的数字顺时针或逆时针数若干个人，

即可知道下一个退出的孩子是谁。

第i个退出的人将获得sigma(i)颗的糖果。

问谁获得的糖果最多（若出现相同取最早退出的人）

算法：

线段树模拟约瑟夫环

代码如下：

#include<cstdio>
#include<iostream>
#include<sstream>
#include<cstdlib>
#include<cstring>
#include<string>
#include<climits>
#include<cmath>
#include<algorithm>
#include<queue>
#include<vector>
#include<stack>
#include<set>
#include<map>
#define INF 0x3f3f3f3f
#define eps 1e-8
using namespace std;

const int MAXN=510000;
int tr[MAXN<<2],a[MAXN];
int d[MAXN];
char s[MAXN][15];

void f() {
for(int i=1; i<MAXN; i++) {
for(int j=1; i*j<MAXN; j++) {
d[i*j]++;
}
}
}

void pushup(int rt) {
tr[rt]=tr[rt<<1]+tr[rt<<1|1];
}

void build(int l, int r, int rt) {
if(l==r) {
tr[rt]=1;
return;
}
int mid=(l+r)>>1;
build(l,mid,rt<<1);
build(mid+1,r,rt<<1|1);
pushup(rt);
}

int query(int l, int r, int k, int rt) {
if(l==r) {
tr[rt]=0;
return l;
}
int ret;
int mid=(l+r)>>1;
if(tr[rt<<1]>=k) {
ret=query(l,mid,k,rt<<1);
} else {
ret=query(mid+1,r,k-tr[rt<<1],rt<<1|1);
}
pushup(rt);
return ret;
}

int main() {
int n,k;
memset(d,0,sizeof(d));
f();
while(scanf("%d%d",&n,&k)==2) {
for(int i=0; i<n; i++) {
scanf("%s%d",s[i],&a[i]);
}
build(0,n-1,1);
int maxn=-1,maxi=1;
for(int i=1; i<=n; i++) {
if(d[i]>maxn) {
maxi=i;
maxn=d[i];
}
}
int nk=k-1;//当前出列位置左边有多少个小盆友
k=query(0,n-1,k,1);
for(int tot=1; tot<maxi; tot++) {
if(a[k]>=0) {
nk=(n-tot+(nk+a[k]-1)%(n-tot))%(n-tot);//n-tot表示队列中还剩多少个人，包括即将出列的那个人
} else {
nk=(n-tot+(nk+a[k])%(n-tot))%(n-tot);
}
k=query(0,n-1,nk+1,1);
}
printf("%s %d\n",s[k],maxn);
}
return 0;
}