Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- class ListNodeIterator:
- def __init__(self, node):
- self.current_node = ListNode(None)
- self.current_node.next = node
- def __next__(self):
- if not self.current_node.next:
- raise StopIteration
- self.current_node = self.current_node.next
- return self.current_node
- class ListNode:
- def __init__(self, value):
- self.value = value
- self.next = None
- def __iter__(self):
- return ListNodeIterator(self)
- def __repr__(self):
- return f"{self.value}"
- list_node = ListNode(1)
- list_node.next = ListNode(2)
- for ln in list_node:
- print(ln)
Add Comment
Please, Sign In to add comment