Surpreso com o bom desempenho da recursão em python
Eu escrevi esta função Python bastante pobre para fatoração principal:
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]
e funcionou como esperado, agora eu estava interessado em saber se o desempenho poderia ser melhor ao usar uma abordagem iterativa:
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
Mas o que observei (enquanto as funções deram os mesmos resultados) foi que a função iterativa levou mais tempo para ser executada. Pelo menos eu tive a sensação, ao fazer isso:
number = 31123478114123
print(factor(number))
print(factor_it(number))
então eu medi:
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))
E é isso que eu tenho:
[290201, 952381, 66666667]
[290201, 952381, 66666667]
[0.19888348945642065]
[0.7451271022307537]
Por que a abordagem recursiva nesse caso é mais rápida que a iterativa?
Eu uso o WinPython-64bit-3.4.4.2 (Python 3.4.4 64bits).