POJ 2154 Color Polya（Polya定理+欧拉函数）

 

题意：将正n边形的n个顶点用n种颜色染色，问有多少种方案（答案mod p，且可由旋转互相得到的算一种）

 

#include <iostream>
#include <stdio.h>
#include <string.h>
#define ll long long
using namespace std;
#define maxn 55555
int prime[maxn];
bool vis[maxn];
int N=0,mod;
void init()
{
 for(int i=2;i<maxn;i++)
 {
     if(!vis[i])
     {
         prime[N++]=i;
         for(int j=i;j<maxn;j+=i)vis[j]=1;

     }
 }
}
int mod_pow(int a,int b)
{
    int res=1;
    a%=mod;
    while(b>0)
    {
        if(b&1)res=(res*a)%mod;
        a=(a*a)%mod;
        b/=2;
    }
    return res;
}
int get_euler(int n)
{
    int ans=n;
    for(int i=0;i<N&&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()
{
    memset(vis,0,sizeof(vis));
    init();
  int t;
  scanf("%d",&t);
  while(t--)
  {
      int n;
      scanf("%d%d",&n,&mod);
      int ans=0;
      for(int i=1;i*i<=n;i++)
      {
          if(n%i==0)
          {
            if(i*i!=n)
                    ans=ans+(get_euler(i)%mod*mod_pow(n,n/i-1)%mod+get_euler(n/i)%mod*mod_pow(n,i-1)%mod)%mod;
                else
                    ans=ans+(get_euler(i)%mod*mod_pow(n,n/i-1)%mod)%mod;
                    ans%=mod;
          }

      }
      printf("%d\n",ans);
  }
    return 0;
}