2

我在实现“埃拉托色尼筛”以获取指定范围内的所有素数时出错。我知道我的代码还没有素数检查,一旦我解决了错误,我会添加它们。

def foo(l,r):
    if l == r:
        return "Error"
    if l > r:
        return "Error"
    if l < r:
        pool = set(xrange(l,r + 1))
        prime = 2
        if l == 1:
            print "Discard 1 for now"
        while prime <= r:
            rem = set(xrange(prime,r + 1,prime))
            pool.difference_update(rem)
            a = 0

            while prime >= pool[a]:
                a = a + 1
            prime = pool[a]
        print pool

foo(1,31623)

错误:

Traceback (most recent call last):
  File "D:\code\sieve_of_eratothenes.py", line 32, in <module>
    foo(1,31623)
  File "D:\code\sieve_of_eratothenes.py", line 27, in foo
    while prime >= pool[a]:
TypeError: 'set' object does not support indexing
4

2 回答 2

2

错误就是它所说的:集合不支持通过索引检索单个项目。看起来您想使用列表或 xrange 对象(例如,pool = xrange(l, r+1)。您为什么使用集合?

请注意,集合中的元素是unordered,因此以您尝试执行的方式对其进行迭代是行不通的。您不能假设较大的素数将位于集合的“末尾”。

于 2012-06-28T04:21:31.120 回答
2

您不能通过索引引用集合元素,但是 aset是可迭代的,因此:

a = 0
while prime >= pool[a]:
   a = a + 1
   prime = pool[a]

可以改写为:

for el in pool:
   if prime >= el:
       prime = el
       break
于 2012-06-28T04:34:15.090 回答