What is the computational complexity of `itertools.combinations` in python?

前端 未结 2 2013
太阳男子
太阳男子 2021-01-12 12:08

itertools.combinations in python is a powerful tool for finding all combination of r terms, however, I want to know about its computational com

2条回答
  •  独厮守ぢ
    2021-01-12 12:32

    I would say it is θ[r (n choose r)], the n choose r part is the number of times the generator has to yield and also the number of times the outer while iterates.

    In each iteration at least the output tuple of length r needs to be generated, which gives the additional factor r. The other inner loops will be O(r) per outer iteration as well.

    This is assuming that the tuple generation is actually O(r) and that the list get/set are indeed O(1) at least on average given the particular access pattern in the algorithm. If this is not the case, then still Ω[r (n choose r)] though.

    As usual in this kind of analysis I assumed all integer operations to be O(1) even if their size is not bounded.

提交回复
热议问题