在 Python 中迭代双端队列的时间复杂度是多少?

2023-12-05

迭代的时间复杂度是多少,或者更准确地说,通过 Python 集合库中的双端队列进行每次迭代的时间复杂度是多少?

一个例子是这样的:

elements = deque([1,2,3,4])
for element in elements:
  print(element)

每次迭代都是一个常数 O(1) 操作吗?或者它是否执行线性 O(n) 操作来获取每次迭代中的元素?

对于所有其他双端队列方法的时间复杂度,网上有很多资源,例如appendleft, append, popleft, pop。似乎没有关于双端队列迭代的任何时间复杂度信息。

Thanks!


如果你的构造是这样的:

elements = deque([1,2,3,4])
for i in range(len(elements)):
    print(elements[i])

You are 不迭代deque,你正在迭代range目的,and then索引到deque。这使得迭代多项式时间缩短,因为每次索引操作,elements[i]是 O(n)。然而,实际上迭代超过deque是线性时间。

for x in elements:
    print(x)

这是一个快速的实证测试:

import timeit
import pandas as pd
from collections import deque

def build_deque(n):
    return deque(range(n))

def iter_index(d):
    for i in range(len(d)):
        d[i]

def iter_it(d):
    for x in d:
        x

r = range(100, 10001, 100)

index_runs = [timeit.timeit('iter_index(d)', 'from __main__ import build_deque, iter_index, iter_it; d = build_deque({})'.format(n), number=1000) for n in r]
it_runs = [timeit.timeit('iter_it(d)', 'from __main__ import build_deque, iter_index, iter_it; d = build_deque({})'.format(n), number=1000) for n in r]

df = pd.DataFrame({'index':index_runs, 'iter':it_runs}, index=r)
df.plot()

And the resulting plot: enter image description here

现在,我们实际上可以看到迭代器协议是如何实现的deque对象在CPython 源代码:

首先,deque对象本身:

typedef struct BLOCK {
    struct BLOCK *leftlink;
    PyObject *data[BLOCKLEN];
    struct BLOCK *rightlink;
} block;

typedef struct {
    PyObject_VAR_HEAD
    block *leftblock;
    block *rightblock;
    Py_ssize_t leftindex;       /* 0 <= leftindex < BLOCKLEN */
    Py_ssize_t rightindex;      /* 0 <= rightindex < BLOCKLEN */
    size_t state;               /* incremented whenever the indices move */
    Py_ssize_t maxlen;
    PyObject *weakreflist;
} dequeobject;

因此,正如评论中所述,deque是一个“块”节点的双向链表,其中块本质上是一个Python对象指针数组。现在来说说迭代器协议:

typedef struct {
    PyObject_HEAD
    block *b;
    Py_ssize_t index;
    dequeobject *deque;
    size_t state;          /* state when the iterator is created */
    Py_ssize_t counter;    /* number of items remaining for iteration */
} dequeiterobject;

static PyTypeObject dequeiter_type;

static PyObject *
deque_iter(dequeobject *deque)
{
    dequeiterobject *it;

    it = PyObject_GC_New(dequeiterobject, &dequeiter_type);
    if (it == NULL)
        return NULL;
    it->b = deque->leftblock;
    it->index = deque->leftindex;
    Py_INCREF(deque);
    it->deque = deque;
    it->state = deque->state;
    it->counter = Py_SIZE(deque);
    PyObject_GC_Track(it);
    return (PyObject *)it;
}

// ...

static PyObject *
dequeiter_next(dequeiterobject *it)
{
    PyObject *item;

    if (it->deque->state != it->state) {
        it->counter = 0;
        PyErr_SetString(PyExc_RuntimeError,
                        "deque mutated during iteration");
        return NULL;
    }
    if (it->counter == 0)
        return NULL;
    assert (!(it->b == it->deque->rightblock &&
              it->index > it->deque->rightindex));

    item = it->b->data[it->index];
    it->index++;
    it->counter--;
    if (it->index == BLOCKLEN && it->counter > 0) {
        CHECK_NOT_END(it->b->rightlink);
        it->b = it->b->rightlink;
        it->index = 0;
    }
    Py_INCREF(item);
    return item;
}

正如您所看到的,迭代器本质上跟踪块索引、指向块的指针以及双端队列中总项目的计数器。如果计数器达到零,它就会停止迭代,如果没有,它会抓取当前索引处的元素,增加索引,减少计数器,并负责检查是否移动到下一个块。换句话说,双端队列可以表示为 Python 中的列表列表,例如d = [[1,2,3],[4,5,6]],并且它迭代

for block in d:
    for x in block:
        ...
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

在 Python 中迭代双端队列的时间复杂度是多少? 的相关文章

随机推荐