破壊的なスタック反復

Question

これは私のスタック実装です。

class Stack: 
    def __init__(self): 
     self.head = None 
     self.size = 0 

    def push(self, item): 
     node = Node(item) 
     if not self.head: 
      self.head = node 
     else: 
      node.next = self.head 
      self.head = node 
     self.size += 1 

    def pop(self): 
     if self.size == 0: 
      raise ValueError('Popping off an empty stack!') 
     item = self.head.val 
     self.head = self.head.next 
     return item 

    def peek(self): 
     if self.size == 0: 
      raise ValueError('Peeking into an empty stack!') 
     return self.head.val 

    def __iter__(self): 
     return self 

    def __next__(self): 
     if self.head: 
      curr = self.head 
     else: 
      raise StopIteration() 
     self.head = self.head.next 
     return curr.val 

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


if __name__ == '__main__': 
    stack = Stack() 
    stack.push(12) 
    stack.push(13) 
    stack.push(9) 
    for item in stack: 
     print(item) 
    print(stack.peek()) 

この問題は繰り返しです。反復は破壊的であり、反復の終了時にピークを呼び出すとエラーがスローされます。

return self.head.val AttributeError: 'NoneType' object has no attribute 'val' 反復を非破壊的にするにはどうすればよいですか？スタックを反復し

Answer 1

これを実行するための簡単な方法あなたのスタックに別の頭を与える反復処理に使用できます。また、__len__メソッドを追加しました。これはスタックのサイズを返します。それは厳密には必要ではないのですが、

class Stack: 
    def __init__(self): 
     self.head = None 
     self.size = 0 

    def __len__(self): 
     return self.size 

    def push(self, item): 
     node = Node(item) 
     if not self.head: 
      self.head = node 
     else: 
      node.next = self.head 
      self.head = node 
     self.size += 1 

    def pop(self): 
     if self.size == 0: 
      raise ValueError('Popping off an empty stack!') 
     item = self.head.val 
     self.head = self.head.next 
     return item 

    def peek(self): 
     if self.size == 0: 
      raise ValueError('Peeking into an empty stack!') 
     return self.head.val 

    def __iter__(self): 
     self.top = self.head 
     return self 

    def __next__(self): 
     if self.top: 
      curr = self.top 
     else: 
      raise StopIteration() 
     self.top = self.top.next 
     return curr.val 

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

if __name__ == '__main__': 
    stack = Stack() 
    stack.push(12) 
    stack.push(13) 
    stack.push(9) 
    print('Size', len(stack)) 
    for item in stack: 
     print(item) 
    print(stack.peek()) 
    stack.push(42) 
    print('Size', len(stack)) 
    for item in stack: 
     print(item) 

出力

Size 3 
9 
13 
12 
9 
Size 4 
42 
9 
13 
12 

それは、__init__にself.top = Noneを追加するために、おそらく良いアイデアです。そしてそれを先頭のアンダースコアのプライベートネームとしてマークすることもできます：self._top。

timgebはコメントで暗示しているように、このアプローチは少し壊れやすいものです。なぜなら、我々はただ1つのself.topしか持っていないからです。私は受け入れるための

def push(self, item): 
    node = Node(item) 
    if self.head: 
     node.next = self.head 
    self.head = node 
    self.size += 1 

Answer 2

あなたが同じスタック上に複数の反復子を持つことができるため、__iter__はイテレータオブジェクトを返す必要があり、：

class Stack: 
    def __init__(self): 
     self.head = None 
     self.size = 0 

    def push(self, item): 
     node = Node(item) 
     if self.head: 
      node.next = self.head 
     self.head = node 
     self.size += 1 

    def pop(self): 
     if self.size == 0: 
      raise ValueError('Popping off an empty stack!') 
     item = self.head.val 
     self.head = self.head.next 
     return item 

    def peek(self): 
     if self.size == 0: 
      raise ValueError('Peeking into an empty stack!') 
     return self.head.val 

    def __iter__(self): 
     return StackIterator(self.head) 

class StackIterator: 
    def __init__(self, head): 
     self.head = head 

    def __iter__(self): 
     return self 

    def __next__(self): 
     if not self.head: 
      raise StopIteration() 
     item = self.head.val 
     self.head = self.head.next 
     return item 

Answer 3

問題はあなたが違いをしないということですiterableスタック自体とイテレータ__iter__が返されます。 __next__は、そのイテレータで呼び出され、Stackでは呼び出されません。

class StackIterator: 
    def __init__(self, stack): 
     self.head = stack.head 

    def __iter__(self): 
     return self 

    def __next__(self): 
     if not self.head: 
      raise StopIteration 

     current = self.head 
     self.head = self.head.next 

     return current.val 

がに__iter__をStackに__next__を取り除くと調整しなさい：

は、私は、次の解決策を提案

def __iter__(self): 
    return StackIterator(self) 

デモ：

>>> stack = Stack() 
>>> stack.push(12) 
>>> stack.push(13) 
>>> stack.push(9) 
>>> 
>>> for x in stack: 
...  print(x) 
... 
9 
13 
12 
>>> stack.peek() 
9