poj 2886 Who Gets the Most Candies?

http://poj.org/problem?id=2886

　　一道借助线段树找出第k个位置的题。

　　题意是，有n个人，他们手中拿着一张写着不为零的数的牌。开始的时候是从第k个人开始的，数到的人出列，然后从那个人的位置开始数d个人，d是他手上的牌的数字，数到的下一个出列，如此反复。第i个出列的人可以得到F(i)个糖果，F(i)是因数的个数。

　　这题主要问题在计算下一个人的位置时，如果next（也就是这个人手上拿的数）是负数的话，就要调整一下位置，因为这时的-1相当于是0了。然后质因数，就直接预处理一下，其他的都没什么大问题了。没调整负数的情况，wa了一次。。。

代码如下：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cstdlib>

#define lson l, m, rt << 1
#define rson m + 1, r, rt << 1 | 1

using namespace std;

const int maxn = 500005;
int cntFac[maxn], maxCnt[maxn];

void pre() {
    for (int d = 1; d < maxn; d++) {
        for (int i = d; i < maxn; i += d) {
            cntFac[i]++;
        }
    }
    for (int i = 1; i < maxn; i++) {
        maxCnt[i] = (cntFac[maxCnt[i - 1]] >= cntFac[i]) ? maxCnt[i - 1] : i;
    }
//    for (int i = 0; i < 20; i++) {
//        printf("%d : %d %d\n", i, cntFac[i], maxCnt[i]);
//    }
}

int rec[maxn], cnt[maxn << 2];
char name[maxn][12];

void up(int rt) {
    cnt[rt] = cnt[rt << 1] + cnt[rt << 1 | 1];
}

void build(int l, int r, int rt) {
    if (l == r) {
        cnt[rt] = 1;
        return ;
    }
    int m = (l + r) >> 1;

    build(lson);
    build(rson);
    up(rt);
}

int locate(int _p, int l, int r, int rt) { // the k-th person, begin from 1
    if (l == r) {
        cnt[rt] = 0;
        return l;
    }
    int m = (l + r) >> 1, ret;

    if (_p <= cnt[rt << 1]) ret = locate(_p, lson);
    else ret = locate(_p - cnt[rt << 1], rson);
    up(rt);

    return ret;
}

int calNext(int cur, int next, int size) {
    if (next < 0) next++;
    return (((cur + next - 1) % size) + size - 1) % size + 1;
}

int deal(int n, int k) {
    int ret = 0, tmp = 0;

    build(1, n, 1);
    rec[0] = k + 1;
//    printf("%d\n", maxCnt[n]);
    for (int i = 0, endi = maxCnt[n]; i < endi; i++) {
        ret = calNext(ret, rec[tmp], n - i);
        tmp = locate(ret, 1, n, 1);
//        printf("ret %d\n", ret);
//        printf("tmp %d\n", tmp);
//        puts("~~~");
    }

    return tmp;
}

int main() {
    int n, k;

//    freopen("in", "r", stdin);
    pre();
    while (~scanf("%d%d", &n, &k)) {
        for (int i = 1; i <= n; i++) {
            scanf("%s %d", name[i], &rec[i]);
        }
        printf("%s %d\n", name[deal(n, k)], cntFac[maxCnt[n]]);
    }

    return 0;
}

——written by Lyon