【POJ】2886 Who Gets the Most Candies?

#include<cstdio>
#define MAXN 500010
int n,k,ans,cnt;
int prime[]={2,3,5,7,11,13,17};
struct node
{
    char name[12];
    int val;
};
node p[MAXN];
int tree[MAXN<<2];
void AntiPrime(int res,int count,int index,int limit)
{
    if(res<=n)
    {
        if(count>cnt)
        {
            cnt=count;
            ans=res;
        }
        else if(count==cnt&&res<ans)
            ans=res;
        int i,temp;
        for(i=temp=1;i<=limit;i++)
        {
            temp*=prime[index];
            if(temp*res>n)
                break;
            else
                AntiPrime(temp*res,count*(i+1),index+1,i);
        }
    }
}
inline void PushUp(int rt)
{
    tree[rt]=tree[rt<<1]+tree[rt<<1|1];
}
void Build(int L,int R,int rt)
{
    if(L==R)
        tree[rt]=1;
    else
    {
        int mid=(L+R)>>1;
        Build(L,mid,rt<<1);
        Build(mid+1,R,rt<<1|1);
        PushUp(rt);
    }
}
int Update(int x,int L,int R,int rt)
{
    if(L==R)
    {
        tree[rt]=0;
        return L;
    }
    int mid=(L+R)>>1,res;
    if(tree[rt<<1]>=x)
        res=Update(x,L,mid,rt<<1);
    else
        res=Update(x-tree[rt<<1],mid+1,R,rt<<1|1);
    PushUp(rt);
    return res;
}
int Query(int x,int y,int L,int R,int rt)
{
    if(x<=L&&R<=y)
        return tree[rt];
    int mid=(L+R)>>1,res=0;
    if(x<=mid)
        res+=Query(x,y,L,mid,rt<<1);
    if(y>mid)
        res+=Query(x,y,mid+1,R,rt<<1|1);
    return res;
}
int main()
{
    int i,mov;
    while(~scanf("%d%d",&n,&k))
    {
        for(ans=cnt=i=1;i<=n;i++)
            scanf(" %s%d",p[i].name,&p[i].val);
        AntiPrime(1,1,0,19);
        Build(1,n,1);
        k=Update(k,1,n,1);
        for(i=1;i<ans;i++)
        {
            mov=Query(1,k,1,n,1);
            if(p[k].val>0)
                mov+=p[k].val;
            else
                mov+=p[k].val+1;
            mov%=tree[1];
            if(!mov)
                mov=tree[1];
            else if(mov<0)
                mov+=tree[1];
            k=Update(mov,1,n,1);
        }
        printf("%s %d\n",p[k].name,cnt);
    }
    return 0;
}