POJ - 2154 Color polya定理+欧拉

题目链接：点击查看

题意：n个项链的环，可以涂n种颜色，供有多少种方案

题解：

这里的m==n,所以结果问哦 phi(n / i) * n^(i - 1) (n%i==0)

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long ll;
int n,p;
int prime[1000010],len;
int ok[1000100];
void init()
{
	for(int i=2;i<=1000000;i++)
	{
		if(!ok[i]) prime[len++]=i;
		for(int j=0;j<len&&(ll)prime[j]*i<=1000000;j++)
		{
			ok[i*prime[j]]=1;
			if(i%prime[j]==0) break;
		}
	}
}
int phi(int x)
{
	int ans=x;
	for(int i=0;i<len&&(ll)prime[i]*prime[i]<=x;i++)
	{
		if(x%prime[i]==0)
		{
			while(x%prime[i]==0) x=x/prime[i];
			ans=ans-ans/prime[i];
		}
	}
	if(x!=1) ans=ans-ans/x;
	return ans%p; 
}
int ksm(int a,int b,int mod)
{
	int ans=1;
	while(b)
	{
		if(b&1) ans=(ans*a)%mod;
		b>>=1;
		a=(a*a)%mod;
	}
	return ans%p;
}
int main()
{
	init();
	int T;
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d%d",&n,&p);
		int ans=0;
		for(int i=1;i*i<=n;i++)
		{
			if(n%i) continue;
			if(i*i==n) ans+=ksm(n%p,i-1,p)*phi(n/i)%p;
			else ans+=ksm(n%p,i-1,p)*phi(n/i)%p+ksm(n%p,n/i-1,p)*phi(i)%p;
			ans=ans%p;
		}
		printf("%d\n",ans);
	}
	return 0;
}