「POJ 2154」Color

传送门

---

problem

给出一个长度为 $n$ 的项链，有 $n$ 种颜色，请求出有多少种染色方案。

注意：一个项链不一定会用完 $n$ 种颜色，并且旋转后相同的染色方案视作同一种。

答案对 $P$ 取模。

数据范围：$1\le n\le 10^9$，$1\le P\le 30000$。

---

solution

这是一道 Polya 定理的模板题。

首先是 Polya 定理的公式：

$$\frac{1}{|G|}\sum _{i=1}^{|G|}{m}^{k_i}$$

其中 $|G|$ 表示置换的数目，$m$ 表示可选颜色数目，$k_i$ 表示第 $i$ 个置换包含的循环个数。

那么对于这道题，由于只有旋转，置换有 $n$ 种。对于第 $i$ 个置换（即旋 $i$ 个珠子），循环的个数是 $\gcd (n,i)$，所以推出来的公式是 $ans=\frac{1}{n}{\sum }_{i=1}^n{n}^{\gcd (n,i)}$。

如果直接枚举 $n$ 计算答案，时间复杂度是 $O(n\log n)$ 的，要优化。

然后就是推式子的过程了：

$$\begin{array}{rl}ans& =\frac{1}{n}\sum _{i=1}^n{n}^{\gcd (i,n)}\\ & =\frac{1}{n}\sum _{k|n}{n}^k\sum _{i=1}^n[\gcd (i,n)=k]\\ & =\sum _{k|n}{n}^{k-1}\sum _{i=1}^{\frac{n}{k}}[\gcd (i,\frac{n}{k})=1]\\ & =\sum _{k|n}{n}^{k-1}\phi (\frac{n}{k})\end{array}$$

我们可以在 $O(\sqrt{n})$ 的时间一个数的 $\phi$ 值（根据公式 $\phi (n)=n{\prod }_{i=1}^k(1-\frac{1}{p_k})$。）

然后我们就可以通过这道题了。

---

code

#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 100005
using namespace std;
int n,P;
int add(int x,int y)  {return x+y>=P?x+y-P:x+y;}
int dec(int x,int y)  {return x-y< 0?x-y+P:x-y;}
int mul(int x,int y)  {return 1ll*x*y%P;}
int power(int a,int b){
	int ans=1;
	for(;b;b>>=1,a=mul(a,a))  if(b&1)  ans=mul(ans,a);
	return ans;
}
int sum,prime[N],mark[N];
void linear_sieves(){
	for(int i=2;i<N;++i){
		if(!mark[i])  prime[++sum]=i;
		for(int j=1;j<=sum&&i*prime[j]<N;++j){
			mark[i*prime[j]]=1;
			if(i%prime[j]==0)  break;
		}
	}
}
int phi(int n){
	int ans=n;
	for(int i=1;prime[i]*prime[i]<=n;++i){
		if(n%prime[i]==0){
			ans=ans/prime[i]*(prime[i]-1);
			while(n%prime[i]==0)  n/=prime[i];
		}
	}
	if(n!=1)  ans=ans/n*(n-1);
	return ans;
}
int main(){
	int T;
	scanf("%d",&T);
	linear_sieves();
	while(T--){
		int ans=0;
		scanf("%d%d",&n,&P);
		for(int i=1;i*i<=n;++i){
			if(n%i==0){
				ans=add(ans,mul(phi(i),power(n,n/i-1)));
				if(i*i!=n)  ans=add(ans,mul(phi(n/i),power(n,i-1)));
			}
		}
		printf("%d\n",ans);
	}
	return 0;
}