http://poj.org/problem?id=2154
poj 2154:Color
http://blog.sina.com.cn/s/blog_6f71bea30100opru.html
大意:n种颜色的珠子可组成多少种长度为n的项链?
这题和2409 类似,不同之处在于,只考虑旋转,不考虑翻转;
因此相对前面两个题目应该说是更简单,但一看数据范围,
就不是这么回事了,2409完全可以直接循环处理,
但这题目n最大达100000000,显然会TLE,故需寻求更佳的解决方案。
用欧拉函数进行优化:
旋转:顺时针旋转i格的置换中,循环的个数为gcd(i,n),每个循环的长度L为n/gcd(i,n)。
如果枚举旋转的格数i,复杂度显然较高。有没有好方法呢?可以不枚举i,反过来枚举L。
由于L|N,枚举了L,再计算有多少个i使得0<=i<=n-1并且gcd(i,n)=n/L。
不妨设a=n/L=gcd(i, n),
不妨设i=a*t则当且仅当gcd(L,t)=1时
Gcd(i,n)=gcd(a*L,a*t)=a。
因为0<=i<n,所以0<=t<n/a=L.
所以满足这个条件的t的个数为Euler(L).
#include<stdio.h>
#include<string.h>
const int N = 36000;
int prime[N];
int pnum;
bool isp[N];
int mod;
void getPrim()
{
int i,j;
pnum = 0;
memset(isp,true,sizeof(isp));
for(i=2;i<N;i++)
if(isp[i])
{
prime[pnum++]=i;
for(j=i+i;j<N;j+=i)
isp[j]=false;
}
// for(i=0;i<10;i++)
// printf("%d ",prime[i]);
}
int Eular(int n)
{
int i;
int ret = n;
if(n==1)return 1;
for(i=0;prime[i]*prime[i]<=n;i++)
{
if(n%prime[i]==0)
{
ret-=ret/prime[i];
while(n%prime[i]==0)n/=prime[i];
if(n==1)break;
}
}
if(n!=1)ret-=ret/n;
return ret;
}
int getPow(int a,int l,int mod)
{
int ans = 1;
a=a%mod;
while(l)
{
if(l&1)ans=(ans*a)%mod;
l>>=1;
a=(a*a)%mod;
}
return ans;
}
int main()
{
getPrim();
int T;
scanf("%d",&T);
{
while(T--)
{
int a,l;
int ans = 0;
scanf("%d%d",&a,&mod);
for(l=1;l*l<a;l++)
{
if(a%l==0)
{
int x = Eular(l)%mod;
int c = getPow(a,a/l-1,mod);
ans=(ans+c*x%mod)%mod;
x = Eular(a/l)%mod;
c = getPow(a,l-1,mod);
ans=(ans+c*x%mod)%mod;
}
}
if(l*l==a)
{
int x = Eular(l)%mod;
int c = getPow(a,a/l-1,mod);
ans=(ans+c*x%mod)%mod;
}
printf("%d\n",ans);
}
}
return 0;
}