在编码比赛中,我们会遇到如下输入:
2 3
4 5
因此,我们执行以下操作:
m, n = [int(x) for x in raw_input().split(' ')]
有没有更快的方法来做同样的事情?
在编码比赛中,我们会遇到如下输入:
2 3
4 5
因此,我们执行以下操作:
m, n = [int(x) for x in raw_input().split(' ')]
有没有更快的方法来做同样的事情?
出于所有实际目的,这几乎是你能得到的最快速度。在某些机器上,您可能会看到订单的加速,或者如果您使用map
而不是列表推导式,您可能会看到几个百分点,但这并不能保证。
这是我机器上的一些快速计时:
from itertools import imap
#map
>>> timeit.timeit('x,y = map(int,line.split(" "))','from __main__ import line')
4.7857139110565186
>>> timeit.timeit('x,y = map(int,line.split())','from __main__ import line')
4.5680718421936035
#list comprehension
>>> timeit.timeit('x,y = [int(x) for x in line.split(" ")]','from __main__ import line')
4.3816750049591064
>>> timeit.timeit('x,y = [int(x) for x in line.split()]','from __main__ import line')
4.3246541023254395
#itertools.imap
>>> timeit.timeit('x,y = imap(int,line.split(" "))','from __main__ import line,imap')
4.431504011154175
>>> timeit.timeit('x,y = imap(int,line.split())','from __main__ import line,imap')
4.3257410526275635
#generator expression
>>> timeit.timeit('x,y = (int(x) for x in line.split(" "))','from __main__ import line')
4.897794961929321
>>> timeit.timeit('x,y = (int(x) for x in line.split())','from __main__ import line')
4.732620000839233
令人惊讶的是,split()
似乎比split(" ")
.
如果您保证有 0 到 9 之间数字的 ascii 输入,则可以使用以下方法做得更好ord
:
>>>timeit.timeit('x,y = [ord(x)-48 for x in line.split(" ")]','from __main__ import line')
1.377655029296875
>>> timeit.timeit('x,y = [ord(x)-48 for x in line.split()]','from __main__ import line')
1.3243558406829834
但这对您的输入施加了严格的限制。
您可以尝试另一种想法(我不知道性能影响会是什么),但您可以从以下位置读取您的行sys.stdin
:
import sys
for line in sys.stdin:
x,y = [ord(x)-48 for x in line.split()]
使用map()
,与内置函数一起使用时,它比列表推导更快:
m, n = map(int, raw_input().split())