-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathbasic_stack.py
More file actions
47 lines (38 loc) · 1.21 KB
/
basic_stack.py
File metadata and controls
47 lines (38 loc) · 1.21 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
from node import Node
class Stack:
def __init__(self, limit=1000):
self.top_item = None
self.size = 0
self.limit = limit
def push(self, value):
if self.has_space():
item = Node(value)
item.set_next_node(self.top_item)
self.top_item = item
self.size += 1
print("Adding {} to the stack!".format(value))
else:
print("No room in stack for {}!".format(value))
def pop(self):
if not self.is_empty():
item_to_remove = self.top_item
self.top_item = item_to_remove.get_next_node()
top_item_value = item_to_remove.get_value()
self.size -= 1
print("{} has been popped from the stack.".format(top_item_value))
return item_to_remove.get_value()
print("Nothing remains in the stack.")
def peek(self):
if not self.is_empty():
return self.top_item.get_value()
print("Nothing currently in the stack.")
def has_space(self):
return self.limit > self.size
def is_empty(self):
return self.size == 0
# Testing the stack class
bts_members = Stack(8)
members = ["Yoongi", "Taehyung", "Namjoon", "Jungkook", "Hoseok", "Jin", "Jimin", "Bang PD"]
for member in members:
bts_members.push(member)
bts_members.pop()