pku2886 Who Gets the Most Candies?

线段树解约瑟夫环，求第i个出圈的人

反素数打的是discuss里的表

 

 

#include <iostream>
#include <algorithm>
using namespace std;

#define MAXN 500001

const int antiprime[] = {1,2,4,6,12,24,36,48,60,120,180,240,360,720,840,1260,1680,2520,5040,7560,10080,15120,20160,25200,27720,45360,50400,55440,83160,110880,166320, 221760, 277200, 332640, 498960, 554400, 665280};
const int factorNum[] = {1, 2, 3, 4, 6, 8, 9, 10, 12, 16, 18, 20, 24, 30, 32, 36, 40, 48, 60, 64, 72, 80, 84, 90, 96, 100, 108, 120, 128, 144, 160, 168, 180, 192, 200, 216, 224};


struct Node{
    int l,r,c;
}nod[3*MAXN];

int next[MAXN],n,k,kk;
char name[MAXN][12];

void buildtree(int u,int l,int r){
    nod[u].c=r-l+1;
    nod[u].l=l;
    nod[u].r=r;
    if(l==r)
        return;
    buildtree(2*u,l,(l+r)/2);
    buildtree(2*u+1,(l+r)/2+1,r);
}


void insert(int u,int k){
    nod[u].c--;
    if(nod[u].l==nod[u].r){
        kk=nod[u].l;
        return;
    }
    if(nod[2*u].c>=k)
        insert(2*u,k);
    else
        insert(2*u+1,k-nod[2*u].c);
}

int query(int u,int l,int r){
    if(l<=nod[u].l && nod[u].r<=r)
        return nod[u].c;
    int res=0;
    if(l<=nod[2u].r)
        res+=query(2*u,l,r);
    if(r>=nod[2*u].l)
        res+=query(2*u+1,l,r);
    return res;
}




int main(){
    int i,ith,tag,&nn=nod[1].c;
    while(scanf("%d%d",&n,&k)!=EOF){
        for(i=1;i<=n;i++)
            scanf("%s%d",&name[i],&next[i]);
        buildtree(1,1,n);

        tag=0;
        while(antiprime[tag]<=n)//返回antiprime中小于等于n的最大反素数下标tag
            tag++;
        tag--;
        ith=antiprime[tag];//第ith个退出的人得到最多的糖

        
        kk=0;
        next[kk]=0;
        while(ith--){
            if(next[kk]>0)
                k=((k+next[kk]-2)%nn+nn)%nn+1;
            else
                k=((k+next[kk]-1)%nn+nn)%nn+1;
            insert(1,k);
        }
        printf("%s %d\n",name[kk],factorNum[tag]);
    

    }
    return 0;
}