能否在单链表__str__方法中复用__iter__且不使用try-except？

问题需求 #

我想要为单链表类实现__str__方法，要求复用已实现的__iter__方法，但不想使用try-except捕获StopIteration——相关资料显示捕获异常时开销较高。同时有疑问：for循环调用__iter__时同样会触发StopIteration，这会不会影响实现？

解决方案 #

可以直接用for循环遍历链表实例（因为__iter__已实现，实例本身是可迭代对象），for循环会自动处理StopIteration，无需手动捕获，既复用了__iter__的遍历逻辑，又避免了手动捕获异常的开销。

修改后的__str__方法 #

def __str__(self):
    values = []
    for node in self:
        values.append(str(node.val))
    return "->".join(values)

原理说明 #

• for循环遍历可迭代对象时，会自动调用__iter__获取迭代器，持续调用next()直到触发StopIteration，此时for循环会自动终止，无需开发者手动处理该异常。
• Python内部对for循环的异常处理做了优化，这种方式的开销远低于手动用try-while-next捕获异常的方式。

完整代码示例 #

from typing import Optional

class ListNode:
    def __init__(self, val=0, next=None):
        self.val = val
        self.next = next

    def __str__(self):
        return f"Node:{self.val} Next:{self.next.val if self.next else 'None'}\n"

class SinglyLinkedList():
    def __init__(self):
        self.head = None
        self.tail = None

    def __iter__(self):
        node = self.head
        while node:
            yield node 
            node = node.next

    def add(self, node):
        if not self.head:
            self.head = node 
        else:
            self.tail.next = node
        self.tail = node

    def __str__(self):
        values = []
        for node in self:
            values.append(str(node.val))
        return "->".join(values)

# 测试代码
one = ListNode(1)
two = ListNode(2)
three = ListNode(3)
four = ListNode(4)
five = ListNode(5)

ll = SinglyLinkedList()
ll.add(one)
ll.add(two)
ll.add(three)
ll.add(four)
ll.add(five)
print(ll)  # 输出：1->2->3->4->5

与其他实现方式对比 #

• 原try-except实现：虽可行，但需手动捕获StopIteration，捕获异常时存在额外性能开销。
• 直接遍历节点的while循环：未复用__iter__逻辑，若后续__iter__的遍历逻辑修改，__str__需同步调整，维护性较差。

内容的提问来源于stack exchange，提问作者heretoinfinity