interactive-coding-challenges/stacks_queues/stack/stack.py
Donne Martin 4cdb85e22d Polish stack challenge and solution (#69)
Update constraints and code.
2016-06-19 20:02:53 -04:00

27 lines
558 B
Python

class Node(object):
def __init__(self, data, next=None):
self.data = data
self.next = next
class Stack(object):
def __init__(self, top=None):
self.top = top
def push(self, data):
self.top = Node(data, self.top)
def pop(self):
if self.top is None:
return None
data = self.top.data
self.top = self.top.next
return data
def peek(self):
return self.top.data if self.top is not None else None
def is_empty(self):
return self.peek() is None