gpt4 book ai didi

python - 计算第一个三角形数在python中有超过500个除数

转载 作者:太空狗 更新时间:2023-10-30 01:02:30 25 4
gpt4 key购买 nike

我正在尝试解决 Project Euler 的第 12 个问题。我可以在将近 4 分钟内计算出除数超过 500 的数字。我怎样才能让它更快?这是尝试;

import time

def main():
memo={0:0,1:1}
i=2
n=200
while(1):
if len(getD(getT(i)))>n:
break
i+=1
print(getT(i))

#returns the nth triangle number
def getT(n):
if not n in memo:
memo[n]=n+getT(n-1)
return memo[n]

#returns the list of the divisors
def getD(n):
divisors=[n]
for i in xrange(1,int((n/2)+1)):
if (n/float(i))%1==0:
divisors.append(i)
return divisors

startTime=time.time()
main()
print(time.time()-startTime)

最佳答案

您不需要数组来存储三角形数。您可以使用单个 int,因为您只检查一个值。此外,使用三角形编号公式可能会有所帮助:n*(n+1)/2 您可以在其中找到第 n 个三角形编号。

getD 也只需要返回一个数字,因为您只是在寻找 500 个除数,而不是除数的值。

但是,您真正的问题在于for 循环中的n/2。通过检查因子对,您可以使用 sqrt(n)。所以只检查小于等于 sqrt(n) 的值。如果检查到 n/2,您会得到大量浪费的测试(数百万)。

所以你要执行以下操作(n 是要查找除数的整数,d 是可能的除数):

  • 确保n/d没有余数。
  • 确定是将 1 还是 2 添加到除数中。

关于python - 计算第一个三角形数在python中有超过500个除数,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/15735028/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com