在不构造列表的情况下查找最多列表

时间:2014-04-20 13:28:11

标签: python

我一直在玩Python,为一些竞争问题制作单行解决方案,我遇到了以下问题。我的解决方案可以写成max([f(k) for k in range(n)])(其中f(k)是一个简单的表达式),看起来很棒,但是当n很大时,它会在找到最大值之前构建一个列表,这超出了内存限制。有没有办法找到最大值而不实际构建列表,同时仍然保持一行?我知道以下工作,但如果我想要,我只会使用不同的语言:

best = f(0)
for k in range(n):
  best = max(best, f(k))
return best

1 个答案:

答案 0 :(得分:7)

max()还需要generator

max(f(k) for k in range(n))

这使您无需构建整个列表。

感谢@DanielRoseman指出在Python 2.x中你应该使用xrange()而不是range()。回想一下,range()返回Python 2.x中的列表,而xrange()返回xrange object(常量内存可迭代)。