python - Python 中的双向链表

Question

结果是一个错误，因为超过了时间限制，但我已经提出了 StopIteration...

我认为我的迭代部分做错了，但很难找到错误。测试输出一直在运行，甚至打印出 None 值。它是如何发生的？

class LinkedListIterator:
    def __init__(self, head):
        self.__current = head.get_next()

    def __iter__(self):
        return self

    def __next__(self):
        if self.__current == None:
            raise StopIteration
        else:
            item = self.__current.get_data()
            self.__current = self.__current.get_next()
            return item

这些是我用来运行程序的输入:

my_list = LinkedListDLL()
my_list.add_to_head(1)
print("Contents:", end=" ")
for node in my_list:
    print(node, end=" ")
print()

Answer 1

此代码旨在在到达列表头部时停止迭代。

    if self.__current == None:
        raise StopIteration

但是，您使用一个不同于None 的NodeDLL 对象来表示头部。

您可以保留对头部的引用并检查它:

class LinkedListIterator:
    def __init__(self, head):
        self._head = head
        self._current = head.get_next()

    def __iter__(self):
        return self

    def __next__(self):
        if self._current is self._head:
            raise StopIteration
        else:
            item = self._current.get_data()
            self._current = self._current.get_next()
            return item