POJ 2886 Who Gets the Most Candies（线段树+约瑟夫环）

  1 /*
  2 ** POJ 2886 Who Gets the Most Candies
  3 ** Created by Rayn @@ 2014/05/09
  4 ** 线段树
  5 */
  6 #include <cstdio>
  7 #include <cstring>
  8 #include <cmath>
  9 #include <algorithm>
 10 using namespace std;
 11 const int MAX = 500100;
 12 
 13 struct Child {
 14     char name[12];
 15     int A;
 16 } child[MAX];
 17 
 18 int n, k, tree[MAX<<2], candy[MAX];
 19 
 20 void GetCandy() //计算约数
 21 {
 22     memset(candy, 0, sizeof(candy));
 23     int limit = (int)sqrt(MAX);
 24     for(int i=1; i<limit; ++i)
 25     {
 26         for(int j=i+1; j*i<MAX; ++j)
 27         {
 28             candy[i*j] += 2;
 29         }
 30         candy[i*i]++;
 31     }
 32     /*
 33     for(int i=1; i<=50; ++i)
 34         printf("candy[%d]: %d\n",i, candy[i]);
 35     printf("\n");
 36     //*/
 37 }
 38 void BuildTree(int rt, int l, int r)
 39 {
 40     tree[rt] = r - l + 1;
 41     if(l == r)
 42     {
 43         return ;
 44     }
 45     int mid = (l + r) >> 1;
 46     BuildTree(rt<<1, l, mid);
 47     BuildTree(rt<<1|1, mid+1, r);
 48 }
 49 int Update(int rt, int l, int r, int val)
 50 {
 51     tree[rt]--;
 52     if(l == r)
 53     {
 54         return l;
 55     }
 56     int ans, mid = (l + r) >> 1;
 57     if(val <= tree[rt<<1])
 58         ans = Update(rt<<1, l, mid, val);
 59     else
 60         ans = Update(rt<<1|1, mid+1, r, val-tree[rt<<1]);
 61     tree[rt] = tree[rt<<1] + tree[rt<<1|1];
 62     return ans;
 63 }
 64 int main()
 65 {
 66 #ifdef _Rayn
 67     //freopen("in.txt", "r", stdin);
 68 #endif
 69 
 70     GetCandy();
 71     while(scanf("%d%d", &n, &k) != EOF)
 72     {
 73         memset(tree, 0, sizeof(tree));
 74         BuildTree(1, 1, n);
 75         for(int i=1; i<=n; ++i)
 76         {
 77             scanf("%s%d", child[i].name, &child[i].A);
 78         }
 79         int maxVal = -1, maxPos, pos;
 80         int num = n;
 81         for(int i=1; i<=n; ++i)
 82         {
 83             pos = Update(1, 1, n, k);
 84             num--;
 85             if(candy[i] > maxVal)
 86             {
 87                 maxVal = candy[i];
 88                 maxPos = pos;
 89             }
 90 
 91             if(num == 0)
 92                 break;
 93             /*
 94             ** 因为取模结果总是0~num-1,所以取模要编号k-1，求出结果后再+1。
 95             ** A>0, 因为这个人出去了，那么后面的人的编号都会先减一
 96             ** 所以这里的k要再-1。
 97             ** A<0，因为这个人出去，对前面的人是没有影响的。
 98             */
 99             if(child[pos].A > 0)
100             {
101                 k = ((k-2 + child[pos].A) % num + num) % num + 1;
102             }
103             else
104             {
105                 //为何模两次就对了呢
106                 //k = ((k-1 + child[pos].A + num) % num + 1;
107                 k = ((k-1 + child[pos].A) % num + num) % num + 1;
108             }
109         }
110         printf("%s %d\n", child[maxPos].name, maxVal);
111     }
112     return 0;
113 }