poj 1091 跳蚤 扩展欧几里得+容斥原理

//poj 1091
//sep9
#include <iostream>
using namespace std;
typedef long long ll;

ll n,m,p[128];
int cnt;

void init()
{
cnt=0;
ll mm=m;
for(ll i=2;i*i<=mm;++i){
if(mm%i==0){
p[cnt++]=i;
while(mm%i==0)
mm/=i;
}
}
if(mm>1)
p[cnt++]=mm;
}

ll quick_power(ll a,ll b)
{
ll res=1;
while(b){
if(b&1)
res*=a;
a*=a;
b/=2;
}
return res;
}

int main()
{
scanf("%lld%lld",&n,&m);
init();
ll ans=0;
for(int s=0;s<(1<<cnt);++s){
int num=0;
ll tmp=1;
for(int i=0;i<cnt;++i)
if((s>>i)&1){
++num;
tmp*=p[i];
}
ll tmp_ans=quick_power(m/tmp,n);
if(num%2==0) ans+=tmp_ans;
else ans-=tmp_ans;
}
printf("%lld",ans);
return 0;
}