poj 2886 Who Gets the Most Candies?

题目链接:http://poj.org/problem?id=2886

题目大意:

有n个孩子围成一个环(顺时针从1编号到n),手里各有一张写着非零数的卡片,一开始第k个孩子先出环,如果出环的孩子的卡片上的数num是正数,那么下一个出环的是其右边的第num个孩子,负数则为其左边的第num个.

第p个出环的孩子可以获得F(p)个糖果,F(p)=p的因子数.

求获得最多糖果的孩子是谁和获得的糖果数,如果有多个,输出最先出环的一个.

题目思路:

预处理出1~n的最大值的F()的数.

第一步:通过F()求出第几个出环的孩子获得的糖果最多.

第二步:线段树维护区间中的孩子的个数sum[],用于求当前排名第几的孩子是谁.

代码:

#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <ctype.h>
#include <math.h>
#include <stack>
#include <queue>
#include <map>
#include <set>
#include <vector>
#include <string>
#include <iostream>
#include <algorithm>
using namespace std;

#define ll long long
#define ls rt<<1
#define rs ls|1
#define lson l,mid,ls
#define rson mid+1,r,rs
#define middle (l+r)>>1
#define eps (1e-8)
#define clr_all(x,c) memset(x,c,sizeof(x))
#define clr(x,c,n) memset(x,c,sizeof(x[0])*(n+1))
#define MOD 1000000009
#define INF 0x3f3f3f3f
#define PI (acos(-1.0))
#define _mod(x,y) ((x)>0? (x)%(y):((x)%(y)+(y))%(y))
#define _max(x,y) (((x)>(y))? (x):(y))
#define _min(x,y) (((x)<(y))? (x):(y))
#define _abs(x) ((x)<0? (-(x)):(x))
#define getmin(x,y) (x= ((x)<0 || (y)<(x))? (y):(x))
#define getmax(x,y) (x= ((y)>(x))? (y):(x))
template <class T> void _swap(T &x,T &y){T t=x;x=y;y=t;}
int TS,cas=1;
const int M=500000+5;
int n,k;
int F[M],sum[M<<2];
struct node{
	char name[11];
	int num;
	void read(){scanf("%s%d",name,&num);}
}p[M];

void build(int l,int r,int rt){
	sum[rt]=r-l+1;
	if(l==r) return;
	int mid=middle;
	build(lson),build(rson);
}

int query(int l,int r,int rt,int p){
	sum[rt]--;
	if(l==r) return l;
	int mid=middle;
	if(sum[ls]>p) return query(lson,p);
	else return query(rson,p-sum[ls]);
}

void show(int l,int r,int rt){
	if(l==r){printf("%d ",sum[rt]);return;}
	int mid=middle;
	show(lson),show(rson);
}

void run(){
	int i,j;
	int mmax=1,pos=1;
	for(i=1;i<=n;i++){
		if(mmax<F[i]) mmax=F[i],pos=i;
		p[i-1].read();
	}
	build(0,n-1,1);
	k--;
	while(pos--){
		i=query(0,n-1,1,k);
		if(!pos || !sum[1]){
			printf("%s %d\n",p[i].name,mmax);
			return;
		}
		if(p[i].num>0) k=_mod(k-1+p[i].num,sum[1]);
		else k=_mod(k+p[i].num,sum[1]);
	}
}

void preSof(){
	for(int i=1;i<M;i++)
		F[i]=1;
	for(int i=1;2*i<M;i++)
		for(int j=2*i;j<M;j+=i)
			F[j]++;
}

int main(){
    //freopen("input.txt","r",stdin);
    //freopen("output.txt","w",stdout);
    preSof();
    //run();
    while(~scanf("%d%d",&n,&k)) run();
    //for(scanf("%d",&TS);cas<=TS;cas++) run();
    return 0;
}