Sorprendido por el buen rendimiento de recursividad en Python
Escribí esta función de Python bastante pobre para la factorización prima:
import math
def factor(n):
for i in range(2, int(math.sqrt(n)+1)):
if not n % i:
return [i] + factor(n//i)
return [n]
y funcionó como se esperaba, ahora estaba interesado en saber si el rendimiento podría ser mejor al usar un enfoque iterativo:
def factor_it(n):
r = []
i = 2
while i < int(math.sqrt(n)+1):
while not n % i:
r.append(i)
n //= i
i +=1
if n > 1:
r.append(n)
return r
Pero lo que observé (mientras que las funciones dieron los mismos resultados) fue que la función iterativa tardó más en ejecutarse. Al menos tuve la sensación, al hacer esto:
number = 31123478114123
print(factor(number))
print(factor_it(number))
entonces medí:
setup = '''
import math
def factor(n):
for i in range(2, int(math.sqrt(n)+1)):
if not n % i:
return [i] + factor(n//i)
return [n]
def factor_it(n):
r = []
i = 2
while i < int(math.sqrt(n)+1):
while not n % i:
r.append(i)
n //= i
i +=1
if n > 1:
r.append(n)
return r
'''
import timeit
exec(setup)
number = 66666667*952381*290201
print(factor(number))
print(factor_it(number))
print(timeit.Timer('factor('+str(number)+')',setup=setup).repeat(1,1))
print(timeit.Timer('factor_it('+str(number)+')',setup=setup).repeat(1,1))
Y esto es lo que obtuve:
[290201, 952381, 66666667]
[290201, 952381, 66666667]
[0.19888348945642065]
[0.7451271022307537]
¿Por qué el enfoque recursivo en este caso es más rápido que el iterativo?
Yo uso WinPython-64bit-3.4.4.2 (Python 3.4.4 64bits).