POJ 2154 Color 【polya+dfs】

看来有必要变得更加强大啊......... 141MS

#include<cstdio>#include<vector>#include<cmath>using namespace std;const int maxn=1<<15;vector<pair<int,int> >v;int vis[maxn],prime[maxn];int n,mod,cnt=0;void init(){    for(int i=2;i<maxn;i++)        if(!vis[i])        {            prime[cnt++]=i;            for(int j=i*i;j<maxn;j+=i)                vis[j]=1;        }}int phi(int x){    int a=x;    for(int i=0;i<cnt;i++)    {        int &p=prime[i];        if(p*p>x) break;        if(x%p==0)        {            while(x%p==0) x/=p;            a-=a/p;        }    }    if(x>1) a-=a/x;    return a%mod;}int pow(int x){    int ans=1,base=n%mod;    while(x)    {        if(x&1) ans=ans*base%mod;        base=base*base%mod;x>>=1;    }    return ans;}int dfs(int pos,int cur){    int ans=phi(n/cur)*pow(cur-1)%mod;    while(pos<v.size())    {        int nxt=cur;        for(int i=0;i<v[pos].second;i++)        {            nxt*=v[pos].first;            ans=(ans+dfs(pos+1,nxt))%mod;        }        pos++;    }    return ans;}void read(int &d){    char ch;    while(ch=getchar(),ch<48||ch>57);d=ch-48;    while(ch=getchar(),ch>47&&ch<58) d=d*10+ch-48;}void write(int d){    if(d==0){puts("0");return;}    char str[6];str[5]=0;int i;    for(i=4;d;i--)        str[i]=d%10+48,d/=10;    puts(str+i+1);}int main(){    int t;init();    read(t);    while(t--)    {        read(n),read(mod);        v.clear();int x=n;        for(int i=0;i<cnt;i++)        {            int num,&p=prime[i];            if(p*p>x) break;            if(x%p) continue;            for(num=0;x%p==0;num++) x/=p;            v.push_back(make_pair(p,num));        }        if(x>1) v.push_back(make_pair(x,1));        write(dfs(0,1));    }    return 0;}