python3 + pypy 29行代码寻找完全数

stripe-python

于 2021-07-08 16:18:51 发布

阅读量220

点赞数

分类专栏： python 文章标签： python jit pypy 数学算法

本文链接：https://blog.csdn.net/weixin_38805653/article/details/118575331

版权

python 专栏收录该内容

70 篇文章 4 订阅

订阅专栏

文章目录

了解完全数
分析算法
编写代码
解决速度问题

了解完全数

如果一个数恰好等于它的真因子之和，则称该数为“完全数” 。各个小于它的约数（真约数,列出某数的约数，去掉该数本身，剩下的就是它的真约数）的和等于它本身的自然数叫做完全数（Perfect number），又称完美数或完备数。
例如：第一个完全数是6，它有约数1、2、3、6，除去它本身6外，其余3个数相加，1+2+3=6。第二个完全数是28，它有约数1、2、4、7、14、28，除去它本身28外，其余5个数相加，1+2+4+7+14=28。第三个完全数是496，有约数1、2、4、8、16、31、62、124、248、496，除去其本身496外，其余9个数相加，1+2+4+8+16+31+62+124+248=496。后面的完全数还有8128、33550336等等。

分析算法

定义factor函数，返回number的真因子之和；
定义judge函数，判断number是否为完全数；
while True循环，暴力列举。

编写代码

import time
def factor(number):
    result = set()
    for j in range(1, number):
        if number % j == 0:
            result.add(j)
            result.add(number // j)
    result.remove(number)
    return result
def judge(number):
    if number == 6:
        return True
    if number < 10:
        return False
    if str(number)[-1] not in ('6', '8'):
        return False
    if str(number)[-1] == '8' and str(number)[-2] != '2':
        return False
    return sum(factor(number)) == number
i = 6
beautiful = 0
begin = time.time()
while True:
    if judge(i):
        beautiful += 1
        print(f'已发现第{beautiful}个完全数,是{i},用时{round(time.time() - begin, 2)}')
    if i % 5000 == 0:
        print(f'已寻找至{i}')
    i += 1

解决速度问题

如果你用上述代码执行，会慢的像乌龟……
这里提供两种方案:

方案一: numba

pip install numba

将上代码改为:

import time
import numba as nb
@nb.jit()
def factor(number):
    result = set()
    for j in range(1, number):
        if number % j == 0:
            result.add(j)
            result.add(number // j)
    result.remove(number)
    return result
@nb.jit()
def judge(number):
    if number == 6:
        return True
    if number < 10:
        return False
    if str(number)[-1] not in ('6', '8'):
        return False
    if str(number)[-1] == '8' and str(number)[-2] != '2':
        return False
    return sum(factor(number)) == number
i = 6
beautiful = 0
begin = time.time()
while True:
    if judge(i):
        beautiful += 1
        print(f'已发现第{beautiful}个完全数,是{i},用时{round(time.time() - begin, 2)}')
    if i % 5000 == 0:
        print(f'已寻找至{i}')
    i += 1