python - 巨大的输入 - for 循环比列表理解更快

Question

我正在尝试解决一个 codechef 初学者问题 -巨大的输入测试。我的代码

a,b = [ int(i) for i in raw_input().split()]
print [input()%b==0 for i in range(a)].count(True)

超时。另一种使用基本 for 循环的解决方案似乎运行良好。

我相信列表理解比基本的 for - 循环更快。那为什么前者慢呢？在这种情况下使用生成器也会减少使用的内存并更快地执行计算，如果是这样我该怎么做？

Answer 1

为什么你认为列表理解比基本for循环更快？（提示：它们都使用相同的底层指令实现。）

您的代码将以如下方式执行：

a, b = ...
temp = []
for i in range(a):
    temp.append(int(raw_input()) % b == 0)
print temp.count(True)

如您所见，它在内存中创建了一个大列表，对其进行迭代以创建第二个列表，然后对第二个列表进行迭代以创建计数。永远不需要创建该列表。

a, b = ...
count = 0
for i in xrange(a):
    if int(raw_input()) % b == 0:
        count += 1
print count

一些编译器能够优化 hylomorphisms 以删除中间列表，但我知道没有 Python 实现能够做到这一点。所以你被困在手动优化中。

注意：不要input在 Python 2.x 中使用，除非您知道自己在做什么。我已经更改了要使用的代码，int(raw_input())因为那是安全的，而这input()是危险的。