我正在尝试解决codechef初学者问题 - Enormous Input Test。我的代码
a,b = [ int(i) for i in raw_input().split()]
print [input()%b==0 for i in range(a)].count(True)
超时了。使用基本for循环的另一个solution似乎工作正常。
我认为列表理解比 - 循环更快。那为什么前者慢了?在这种情况下也会使用生成器减少使用的内存并更快地执行计算,如果是这样我该怎么办?
答案 0 :(得分:5)
为什么您认为列表理解比基本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实现能够做到这一点。所以你手动优化了。
注意:不要在Python 2.x中使用input
,除非您知道自己在做什么。我已将代码更改为使用int(raw_input())
,因为这是安全的,而input()
则很危险。