POJ 2480 Longge's problem (欧拉函数)

题意：求Σgcd(i,N)。

思路：若gcd(A,N)=i，那么gcd(A/i,N/i)=1，因此，对于N的每一个因子，sum+=φ(N/i)*i。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<map>
#include<queue>
#include<stack>
#include<set>
#include<cmath>
#include<vector>
#define inf 0x3f3f3f3f
#define Inf 0x3FFFFFFFFFFFFFFFLL
#define eps 1e-9
#define pi acos(-1.0)
using namespace std;
typedef long long ll;
const int maxn=50000+10;
int primes[maxn],pcnt;
int factor[110],num[110],tot,n;
ll sum;
bool flag[maxn];
void getprimes()
{
    pcnt=0;
    memset(flag,0,sizeof(flag));
    for(int i=2;i<maxn;++i)
    {
        if(!flag[i])
        {
            primes[pcnt++]=i;
            for(ll j=(ll)i*i;j<maxn;j+=i)
                flag[j]=true;
        }
    }
}
int euler_phi(int n)
{
    int ans=n;
    int m=sqrt(n+0.5);
    for(int i=0;i<pcnt&&primes[i]<=m;++i)
    {
        if(n%primes[i]==0)
        {
            ans=ans-ans/primes[i];
            while(n%primes[i]==0) n/=primes[i];
        }
    }
    if(n>1) ans=ans-ans/n;
    return ans;
}
void dfs(int pos,int now)
{
    if(pos==tot)
    {
        sum+=(ll)euler_phi(n/now)*now;
        return ;
    }
    int tmp=1;
    for(int i=0;i<=num[pos];++i)
    {
        dfs(pos+1,now*tmp);
        tmp=tmp*factor[pos];
    }
}
void solve(int n)
{
    tot=0;
    for(int i=0;i<pcnt;++i)
    {
        if(n%primes[i]==0)
        {
            factor[tot]=primes[i];num[tot]=0;
            while(n%primes[i]==0) {n/=primes[i];num[tot]++;}
            tot++;
        }
    }
    if(n>1) {factor[tot]=n;num[tot++]=1;}
    sum=0;
    dfs(0,1);
}
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    getprimes();
    while(cin>>n)
    {
        solve(n);
        cout<<sum<<endl;
    }
    return 0;
}