codeforces 850c sg函数

我们发现每次只能选质数的倍数，而且各个质数之间是互不影响的，我们先把n个a数质因数分解，对于一个质数p，a1-an中可能存在p^1,p^2,p^k这些不同的质因数的指数，这就是一个nim游戏，对于p，先手赢还是后手赢，最后把各个质数的sg函数求抑或就行了。

考场上并没有想到怎样求sg函数，看看题解发现，对于一个质数p，存在1，3，4，..k这些指数，就用二进制存起来表示成一个数，也就是状态压缩，于是求sg函数的时候，对于一个状态x，假设我们此时选择p^i去除，那么他就会被(x>>i)|(x&((1<<(i-1))-1)),因为比p^i大的才会除以p^i，x>>i，小于的就不会变于是就是取并集，(x&((1<<(i-1))-1))就是后i位不变。

x=0时是先手必败态，所以直接返回0;

#include<cstdio>
#include<cstring>
#include<set>
#include<map>
#define maxl 100010

using namespace std;

int n,cnt,ans;
int a[maxl],p[maxl],num[maxl],sum[maxl],dy[maxl];
bool no[maxl];
map <int,int> st,sg;

void shai()
{
	int t,j;
	no[1]=true;
	for(int i=2;i<maxl;i++)
	{
		if(!no[i])
			p[++p[0]]=i,dy[i]=i;
		t=p[1]*i;j=1;
		while(j<=p[0] && t<maxl)
		{
			no[t]=true;
			if(i%p[j]==0)
			{
				dy[i]=p[j];
				break;
			}
			t=p[++j]*i;
		}
	}
}

void prework()
{
	cnt=p[0];st.clear();
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		for(int j=1;j<=p[0];j++)
		{
			int d=0;
			while(a[i]%p[j]==0 && a[i]>1)
				a[i]/=p[j],d++;
			if(d>0)
				st[p[j]]|=1<<(d-1);
		}
		if(a[i]>1)
			st[a[i]]|=1;
	}
}

int getsg(int x)
{
	if(x==0)
		return 0;
	if(sg.count(x)) return sg[x];
	map<int,int> vis;
	int p=x,t=0;
	while(p)
		p>>=1,t++;
	for(int i=1;i<=t;i++)
		vis[getsg((x>>i)|(x&((1<<(i-1))-1)))]=1;
	for(int i=0;i<maxl;i++)
	if(!vis[i])
		return sg[x]=i;
}

void mainwork()
{
	ans=0;
	map<int,int>::iterator it;
	for(it=st.begin();it!=st.end();it++)
		ans=ans^getsg((*it).second);
}

void print()
{
	if(ans)
		printf("Mojtaba\n");
	else
		printf("Arpa\n");
}

int main()
{
	shai();
	while(~scanf("%d",&n))
	{
		prework();
		mainwork();
		print();
	}
	return 0;
}