poj2154Color

polyd计数问题，题意是给你n个珠子n种染料，可以旋转，可以有多少种不等价的染色方案。ps：一定要慎重处理取余的运算，不是素数的mod，个人拙见，如不对请指正。

#include<cstdio>#include<map>#include<vector>#include<algorithm>#include<iostream>using namespace std;typedef long long ll;ll n;int mod;int T;ll mod_pow(ll x,ll n,int MOD){    ll res=1;    while(n>0)    {        if(n&1)res=res*x%MOD;        x=x*x%MOD;        n>>=1;    }    return res;}map<int ,int> prime_factor(int n){    map<int , int>res;    for(int i=2; i*i<=n; i++)    {        while(n%i==0)        {            ++res[i];            n/=i;        }    }    if(n!=1)        res[n]=1;    return res;}vector<int> divisor(int n){    vector<int> res;    for(int i=1; i*i<=n; i++)    {        if(n%i==0)        {            res.push_back(i);            if(i!=n/i)res.push_back(n/i);        }    }    return res;}void solve(){    map<int ,int>primes = prime_factor(n);    vector<int>divs=divisor(n);    ll res=0;    for(int i=0; i<divs.size(); i++)    {        ll euler = divs[i];        for(map<int ,int>::iterator it = primes.begin(); it!=primes.end(); ++it)        {            int p = it->first;            if(divs[i]%p == 0)euler = euler / p * (p - 1);        }        res+=euler*mod_pow(n,n/divs[i]-1,mod)%mod;        res%=mod;    }    printf("%lld\n",res);}int main(){    scanf("%d",&T);    while(T--)    {        scanf("%lld %d",&n,&mod);        solve();    }    return 0;}