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

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

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

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

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

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

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

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

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

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

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

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

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

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

     prime_table();

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

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

     return ;
 }

代码君