【poj2886】Who Gets the Most Candies?——线段树

题目：

题目链接

描述：

有N个小孩围坐成一圈（N<=500000），顺时针从1到N标号，第i个小孩有一个val[i]。多组用例，每次给定N，K，表示从第K个小孩开始，依次退出圆圈，下一个小孩的位置由val[k]指定，val>0时表示从这一次出圈的小孩的逆时针方向第val个，否则顺时针（此时已经不算出圈的小孩）。第p个出圈的小孩可以获得p的约数个糖果。每组用例给出每个小孩的名字和val值，求拿到最多的糖果的小孩的名字及拿到的糖果数。如果有重复情况输出第一个跳出圈的孩子的名字。

题解：

本题时间限制比较严格，所以要先预处理出所有数的约数的个数。
这样给定N之后，就可以从预处理的表格中找到最多的糖果数。
接下来要确定的就是第一个拿到这个糖果数的孩子在原序列中排第几。
用线段树维护区间内还在圈内的孩子的数量，每次更新返回满足要求的位置数的下标即为现序列第k个孩子在原序列的位置。

代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
#define ls t<<1
#define rs t<<1|1
using namespace std;
const int maxn = 500005;
int tree[maxn * 3],val[maxn],pr[maxn];
char name[maxn][20];
void pushup(int t)
{
    tree[t] = tree[ls] + tree[rs];
}
void build(int l,int r,int t)
{
    if(l == r)
    {
        tree[t] = 1;             //初始时每个位置都有孩子
        scanf("%s%d",name[l],&val[l]);
        return;
    }
    build(l,(l+r)/2,ls);
    build((l+r)/2+1,r,rs);
    pushup(t);
}
int update(int l,int r,int t,int pos)
{
    int ret;
    if(l == r)
    {
        tree[t] = 0;
        return l;
    }
    if(pos <= tree[ls]) ret = update(l,(l+r)/2,ls,pos);   //先在左子树找，左子树不够就去右子树，最终定位到叶节点
    else ret = update((l+r)/2+1,r,rs,pos-tree[ls]);
    pushup(t);
    return ret;
}
void pre()
{
    for(int i = 1;i < maxn;i++)
    {
        for(int j = i;j < maxn;j+=i)
        {
            pr[j]++;
        }
    }
}

int main()
{
    //freopen("in.txt","r",stdin);
    int n,k,ans = 0,pos = 1;
    pre();
    while(~scanf("%d%d",&n,&k))
    {
        for(int i = 1;i <= n;i++)
            ans = max(ans, pr[i]);
        build(1,n,1);
        for(int i = n-1;i >= 0;i--)
        {
            pos = update(1,n,1,k);                 //pos为现在第k个孩子在原序列的位置
            if(pr[n-i] == ans || !i)break;         //n-i表示出圈的顺序
            if(val[pos] >= 0) k = (k - 1 - 1 +val[pos]) % i + 1;    //第一个-1表示去掉出圈的孩子，第二个表示转换成以0为起始，最后+1换回以1为起始
            else k = ((k - 1 + val[pos]) % i + i) % i + 1;
        }
        printf("%s %d\n",name[pos], ans);
    }
    return 0;
}