POJ 2886 Who Gets the Most Candies?

对于当前要出去的人k,我们可以知道下一个要出去的人是第几大的

我的做法：统计编号为k+1到n的人中还剩多少人，记为x，编号为1到k-1的还剩多少人，记为y,以及从k开始要往右数几个人,记为t，如果t<=y,那么下一个出去的人就是第y+t大的，否则，为第t-x大的，我们用线段树来查找这第k大的人的编号，然后更新

还有一种方法是先统计k左边的人数x,记k要往右数y个人，且k走后还剩下n个人

如果y>0,next=(x+y-1)%n+1

如果y<0,next=(x+y)%n+1

next表示第几大

一开始打表求因子个数弄搓了，老是TLE,后来换了别人的方法跑了2700多ms

代码：

#include<iostream>
#include<memory.h>
#include<string>
#include<cstdio>
#include<algorithm>
#include<math.h>
#include<stack>
#include<queue>
#include<vector>
#include<map>
#include<ctime>
using namespace std;
const int MAX=500005;
struct node 
{
	int l,r,cnt,lcnt,rcnt;
}t[MAX*4];
char name[MAX][12];
int f[MAX],n,a[MAX],res[MAX],num[MAX];
void init()
{
	int i,j;
	memset(f,0,sizeof(f));
	for(i=1;i<=500000;i++)
		for(j=1;j*i<=500000;j++)
			f[j*i]++;
}
void build(int ll,int rr,int n)
{
	t[n].l=ll; t[n].r=rr; 
	if(ll==rr)
	{
		t[n].lcnt=t[n].rcnt=t[n].cnt=1;
		return;
	}
	int mid=(ll+rr)>>1;
	build(ll,mid,n<<1);
	build(mid+1,rr,n<<1|1);
	t[n].lcnt=t[n<<1].cnt; t[n].rcnt=t[n<<1|1].cnt;
	t[n].cnt=t[n].lcnt+t[n].rcnt;
}
void update(int ll,int rr,int n)
{
	if(t[n].l==t[n].r)
	{
		t[n].lcnt=t[n].rcnt=t[n].cnt=0;
		return;
	}
	int mid=(t[n].l+t[n].r)>>1;
	if(mid>=rr)
		update(ll,rr,n<<1);
	else
		update(ll,rr,n<<1|1);
	t[n].lcnt=t[n<<1].cnt;
	t[n].rcnt=t[n<<1|1].cnt;
	t[n].cnt=t[n].lcnt+t[n].rcnt;
}
int query(int n,int k)
{
	if(t[n].l==t[n].r)
		return t[n].l;
	if(t[n].lcnt>=k)
		return query(n<<1,k);
	else
		return query(n<<1|1,k-t[n].lcnt);
}
int query2(int ll,int rr,int n)
{
	if(t[n].l==ll&&t[n].r==rr)
		return t[n].cnt;
	int mid=(t[n].l+t[n].r)>>1;
	if(mid>=rr)
		return query2(ll,rr,n<<1);
	else if(mid<ll)
		return query2(ll,rr,n<<1|1);
	else
	{
		return query2(ll,mid,n<<1)+query2(mid+1,rr,n<<1|1);
	}
}
int main()
{
	int i,j,k;
	init();
	while(scanf("%d%d",&n,&k)!=EOF)
	{
		for(i=1;i<=n;i++)
		{
			scanf("%s %d",name[i],&a[i]);
		}
		build(1,n,1);
		int t,x,y,del;
		for(i=1;i<n;i++)
		{
			res[k]=f[i];
			num[k]=i;
			//cout<<k<<endl;
			update(k,k,1);
			if(k<n)
			{
		 		x=query2(k+1,n,1);
		 		y=query2(1,k,1);
			}
			else
			{
				x=y=0;
			}
			if(a[k]<0)
				del=1;
			else
				del=0;
			a[k]=((a[k]+del)%(n-i)+n-i)%(n-i);
			if(a[k]==0)
				a[k]+=n-i;
			if(a[k]<=x)
				t=y+a[k];
			else
				t=a[k]-x;
			k=query(1,t);
		}
		k=query(1,1);
		//cout<<k<<endl;
		res[k]=f[n];
		num[k]=n;
		int ans=-1;
		for(i=1;i<=n;i++)
		{
			if(res[i]>ans)
			{
				j=i;
				ans=res[i];
			}
			else if(res[i]==ans&&num[i]<num[j])
				j=i;
		}
		printf("%s %d\n",name[j],ans);
		//cout<<num[j]<<endl;
	}
	return 0;
}