51Nod 1240 莫比乌斯函数

题目链接

#include<iostream>
#include<cstdio>
#include<math.h>
#include<cstring>
#include<algorithm>
#include<queue>
#include<set>
#include<vector>
using namespace std;
int f(int n){
int p=sqrt(n);
if(p*p==n) return 0;
int k=0;
int sum=0;
for(int i=2;i<=p;i++){
if(n%i==0){//保证了每个i都是质数
k=0;
while(n%i==0) n/=i,k++;
if(k>1) return 0;
sum++;
}
}
if(n!=1) sum++;
return pow(-1,sum);
}
int main(){
int n;
cin>>n;
printf("%d",f(n));
return 0;
}