51Nod-1240-莫比乌斯函数

51Nod-1240-莫比乌斯函数

                    1240 莫比乌斯函数  

莫比乌斯函数，由德国数学家和天文学家莫比乌斯提出。梅滕斯(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

---

解题方法

题意为：
1.输出为0，即当n的因子中含有平方因子;
2.输出为-1，即当n的因子不含有平方因子，且质因子个数为奇数个;
3.输出为1，即当n的因子不含有平方因子，且质因子个数为偶数个。

解法：
任意一个数都能由质因子来合成。
遍历n的所有质因子，如果同一个因子出现两次，即含有平方因子，则符合第一种情况，输出为0;
否则，统计质因子出现的个数，然后判断奇偶。

---

解题代码

def miu(n):
    num = 0

    # 开始用质因子分解n
    for i in range(2, n+1):
        cnt = 0
        if i*i > n: break

        if n%i == 0:
            num += 1       # 统计不同质因子个数
            while n%i == 0:
                n //= i    # 每次分解成功都更新n
                cnt += 1   # 统计相同质因子个数
            if cnt >= 2:
                return 0

    return -1 if num%2 == 0 else 1  # 判断奇偶

while True:
    try:
        N = int(input())
        print(miu(N))
    except EOFError:
        break