POJ (线段树) Who Gets the Most Candies?

这道题综合性挺强的，又牵扯到数论，又有线段树。

线段树维护的信息就是区间中有多少个人没跳出去，然后计算出下一个人是剩下的人中第几个。

我在这调程序调了好久，就是那个模来模去的弄得我头晕。

不过题确实是好题，给赞。

#include <cstdio>
#include <iostream>
#include <cmath>
#include <string>
#include <algorithm>
#include <vector>
using namespace std;

const int maxn = 500000 + 10;
int n, k;
int sum[maxn << 2];

void build(int o, int L, int R)
{
    if(L == R) { sum[o] = 1; return; }
    int M = (L + R) / 2;
    build(o*2, L, M);
    build(o*2+1, M+1, R);
    sum[o] = sum[o*2] + sum[o*2+1];
}

int update(int o, int L, int R, int p)
{
    if(L == R) { sum[o] = 0; return L; }
    int M = (L + R) / 2;
    int ans;
    if(sum[o*2] >= p) ans = update(o*2, L, M, p);
    else ans = update(o*2+1, M+1, R, p-sum[o*2]);
    sum[o] = sum[o*2] + sum[o*2+1];
    return ans;
}

const int maxp = 750;
int prime[maxp], pcnt = 0;
bool vis[maxp];

void prime_table()
{
    int m = sqrt(maxp);
    for(int i = 2; i <= m; i++) if(!vis[i])
        for(int j = i * i; j < maxp; j += i) vis[j] = true;
    for(int i = 2; i < maxp; i++) if(!vis[i]) prime[pcnt++] = i;
}

int F(int n)
{
    int ans = 1;
    for(int i = 0; i < pcnt && n > 1; i++) if(n % prime[i] == 0)
    {
        int c = 0;
        while(n % prime[i] == 0) { n /= prime[i]; c++; }
        ans *= c + 1;
    }
    if(n > 1) ans <<= 1;
    return ans;
}

int next[maxn];
char stu[maxn][11];

int main()
{
    //freopen("in.txt", "r", stdin);

    prime_table();

    while(scanf("%d%d", &n, &k) == 2)
    {
        build(1, 0, n - 1);
        int _max = 0, id;
        for(int i = 0; i < n; i++) { scanf("%s%d", stu[i], &next[i]); }

        int pos = k - 1;
        for(int i = 1; i <= n; i++)
        {
            int t = update(1, 0, n - 1, pos + 1);
            int f = F(i);
            if(f > _max)
            {
                _max = f;
                id = t;
            }
            if(i == n) break;
            int MOD = n - i;
            if(next[t] < 0) next[t]++;  //负数的情况还要有一点小改动，在这坑了好久
            pos = (((pos + next[t] - 1) % MOD) + MOD) % MOD;
        }
        printf("%s %d
", stu[id], _max);
    }

    return 0;
}