除了暴力法,
用数论的推导公式进行解题,可以达到AC:
思路参考:(420条消息) Project euler 401 约数的平方和 (数论)_LzyRapX的博客-CSDN博客
以下是python3.8,jupyter实现,详情思路在注释:
#数论公式参考https://blog.csdn.net/liangzhaoyang1/article/details/58269542
import math
#计算f(n) = 1/6*n*(n+1)*(2*n+1)
def f(n):
ans=1/6*n*(n+1)*(2*n+1);
#ans = ans % mod
return ans
#进行计算两个主要式子
#math.ceil()向上取整,math.round()四舍五入,math.floor()向下取整
def calc(n):
ans = 0;#保存两个式子之和
t2 = math.floor(math.sqrt(n));#按题进行向下取整
t1 = math.floor(n / ( t2 + 1 ));#同理
#print(t2,t1)
#计算第一个主要式子k^2*(n/k),的和
for k in range(1,int(t1)+1):
ans = ans + k*k*( math.floor(n/k) );#ans+k^2*(n/k)
#计算第二个主要式子的和,并用中间式子替代后计算
for d in range(1,int(t2)+1):
ans=ans + d*( f(math.floor(n/d)) - f(math.floor(n/(d+1))) );
return ans;
mod = 1e9+7#用来约分
n = int(input())#输入要计算的值
##前6项为:1、6、16、37、63和113
#round为四舍五入
print(round(calc(n)%mod))
# 输入
# 100000
# 样例输出
# 680584257