51nod点头网-1240 莫比乌斯函数

题目链接

1240 莫比乌斯函数
基准时间限制：1 秒 空间限制：131072 KB 分值: 0 难度：基础题 收藏 关注
莫比乌斯函数，由德国数学家和天文学家莫比乌斯提出。梅滕斯(Mertens)首先使用μ(n)（miu(n)）作为莫比乌斯函数的记号。（据说，高斯(Gauss)比莫比乌斯早三十年就曾考虑过这个函数）。
具体定义如下：
如果一个数包含平方因子，那么miu(n) = 0。例如：miu(4), miu(12), miu(18) = 0。
如果一个数不包含平方因子，并且有k个不同的质因子，那么miu(n) = (-1)^k。例如：miu(2), miu(3), miu(30) = -1,miu(1), miu(6), miu(10) = 1。
给出一个数n, 计算miu(n)。
Input
输入包括一个数n，(2 <= n <= 10^9)
OutPut
输出miu(n)。
Input示例
5
Output示例
-1

先进行素数筛选，之后根据素数的平方能否被n整除判断是否包含平方因子，若包含，则输出0；若不包含，则进行素因子分离，并且统计不同的质因子个数，若不同的质因子个数为奇数则输出-1，否则输出1；

#include <iostream>

using namespace std;
const int N=1000010;
int a[N],b[N];
int main()
{  int i,j,k,p,t,sum,flag;
    k=0;
   for(i=0;i<N;i++)
{
    a[i]=1;
}
  for(i=2;i<N;i++)
  {
      if(a[i])
     {   b[k]=i;
         k++;
         for(j=i+i;j<N;j+=i)
         {
             a[j]=0;
         }
     }
  }
       while(cin>>p)
       {   flag=0;
           for(i=0;b[i]*b[i]<=p;i++)
           {
               if(p%(b[i]*b[i])==0)
               {
                   t=0;
                   flag=1;
                   break;
                }
           }
           if(flag==0)
           {  sum=0;
             for(i=0;b[i]*b[i]<=p;i++)
             {
                 if(p%b[i]==0)
                 {
                     sum++;
                     while(p%b[i]==0)
                     {
                         p/=b[i];
                     }
                 }
             }
             if(p>1)  sum++;
                 if((sum&1))
                 t=-1;
                 else t=1;
           }
           cout<<t<<endl;
       }
       return 0;
}