python生成器时间复杂度混乱

2023年5月26日 266次阅读

我一直在阅读有关
python中关键字yield和生成器的内容,我想知道我是否理解它的时间复杂度.

这是我的生成器函数来获取因子：

def calc_factors(n):
    for k in range(0, n+1):     # this is the second "for" loop
        if n % k == 0:
           yield k

我会把这个生成器函数称为：

>>> for factor in calc_factor(100):   # this is the first "for" loop
        print factor

现在我的理解是,它具有O(n ^ 2)的时间复杂度,因为它有两个for循环,但我再次不相信.请在这方面赐教.
提前致谢！

最佳答案 你误会了.你有一个O(n)循环.生成器函数上的循环不是嵌套循环,它只是在生成时从生成器接收每个项目.

换句话说,calc_factor(100)循环中的for因子直接与yield k表达式相关联;每次在calc_factor(100)循环中执行for因子时,前进一步.对于每个执行的yield k表达式,您将获得1个因子值. yield k执行(最多)n次,不再执行.

你可以在不过分夸大事实的情况下,将calc_factor(100)循环体中for因子中的所有代码想象为替换yield k行,其中factor = k.在您的情况下,您只使用打印因子,因此您可以使用打印k替换yield k而不会丢失功能.

如果您想以不同方式查看它,请取走生成器并构建列表.这就是您的代码在这种情况下所做的事情：

results = []
for k in range(0, n+1):
    if n % k == 0:
       results.append(k)

for factor in results:
    print factor

现在你还有两个循环,但它们不是嵌套的.一个是用于构建列表的O(n)循环,另一个是用于打印结果的O(k)循环(具有k

点赞