hdu 3187 （欧拉函数+dfs）

点击打开链接

/*

给出一个数n，满足P(k)=n，其中k的素因子个数<=3；n<2^31;
    欧拉函数：
        phi(k)=k*(1-1/p1)(1-1/p2)(1-1/p3)
              =(p1-1)*p1^x * (p2-1)*p2^y * (p3-1)*p3^z;
    phi(1)=1;
    因为n=phi(k),枚举pi-1，if(n%(pi-1)==0&&is_prime(pi))则加入pi;
    然后用DFS，最大有三个数使phi(k)=n;
    注意phi(1)=1;
    当n==1时，有两种情况，k=1,2；


    注意这里枚举（pi-1），不是n的素因子；

2013/04/22-13:36

*/

#include"stdio.h"
#include"string.h"
#include"algorithm"
using namespace std;
int n,ans;
int A[4010];
int cnt;
int ret;
int t[3];
int is_prime(int x)
{
    int i;
    for(i=2;i*i<=x;i++)
        if(x%i==0)return 0;
        return 1;
}
//从第x个开始，已经找到y个，总共需要找z个。。。
void dfs(int x,int y,int z)
{
    int i;    
    if(y==z)
    {
        int tem=n;
        int f=0;
        for(i=0;i<z;i++)
        {
            if(tem%(t[i]-1)!=0)
            {
                f=1;break;
            }
            tem/=(t[i]-1);
        }
        if(f==0)
        {
            for(i=0;i<z;i++)
            {
                while(tem%t[i]==0)
                    tem/=t[i];
            }
            if(tem==1)ret++;
        }
    }
    else
    {
        for(i=x;i<cnt;i++)
        {
            t[y]=A[i];
            dfs(i+1,y+1,z);
        }
    }
}

int main()
{
    while(scanf("%d",&n)!=-1)
    {
        if(n==1)
        {
            printf("2\n");
            continue;
        }
        else
        {
            int i,j;
            cnt=0;
            //将n的素因子存入A中。
            for(i=1;i*i<=n;i++)
            {
                if(n%i==0&&is_prime(i+1)==1)
                    A[cnt++]=i+1;
                if(n%i==0&&is_prime(n/i+1)==1)
                    A[cnt++]=n/i+1;
            }
            sort(A,A+cnt);
            //可能有重复的。。。
            for(i=1;i<cnt;i++)
            {
                if(A[i]==A[i-1])
                {
                    for(j=i;j<cnt-1;j++)
                        A[j]=A[j+1];
                    cnt--;
                }
            }
            ans=0;
            for(i=1;i<=3;i++)
            {
                ret=0;
                dfs(0,0,i);
                ans+=ret;
            }
            printf("%d\n",ans);
        }
    }
    return 0;
}