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