python - 为什么KeyError : 0

Question

我正在尝试解决 Project Euler 21 但我收到 KeyError: 0 这通常发生在您引用不存在的字典键时。但是，我认为我已经解决了 < 10000 条件下的问题。错误指的是 main() 函数中的第一个“if”语句。

sumsdivs = {}
for i in range(1, 10000):
    tmpls = []
    for j in range(1, i):
        if i % j == 0:
            tmpls.append(j)
    sumsdivs[i] = sum(tmpls)

amls = []
def main():
    for i in range(1, 10000):
        if sumsdivs[i] < 10000 and sumsdivs[i] == sumsdivs[sumsdivs[i]]:
            if sumsdivs[i] not in amls:
                amls.append(sumsdivs[i])
            if sumsdivs[sumsdivs[i]] not in amls:
                amls.append(sumsdivs[sumsdivs[i]])
    return sum(amls)

print(main())

有什么想法吗？

Answer 1

您在此处为 i = 1 插入 0:

sumsdivs = {}
for i in range(1, 10000):
    tmpls = []
    for j in range(1, i):
        if i % j == 0:
            tmpls.append(j)
    sumsdivs[i] = sum(tmpls)

内部循环永远不会运行(range(1, 1) 为空)，并且 sum([]) 为 0。

演示:

>>> sumsdivs = {}
>>> for i in range(1, 2):
...     tmpls = []
...     for j in range(1, i):
...         if i % j == 0:
...             tmpls.append(j)
...     sumsdivs[i] = sum(tmpls)
... 
>>> sumsdivs
{1: 0}
>>> sum([])
0    

因此 sumsdivs[1] 为 0，并且 sumsdivs[sumsdivs[0]] 抛出 KeyError:

>>> sumsdivs[sumsdivs[1]]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
KeyError: 0