3 栈
栈是一种数据结构,只能从一端插入和删除操作,遵循着先进后出原则存储数据。
3.1 栈的初始化#
def __init__(self):
self.stack = [] # 栈列表
self.size = 20 # 栈大小
self.top = -1 # 栈顶位置
3.2 元素进栈#
# 元素进栈
def push(self, element):
self.stack.append(element)
self.top += 1
3.3 元素出栈#
# 元素出栈
def pop(self):
element = self.stack[-1]
self.top -= 1
del self.stack[-1]
return element
这里可以直接调用pop函数,使用如下:
self.stack.pop() # 弹出栈顶元素
3.4 获取栈顶元素#
# 获取栈顶位置
def getTop(self):
return self.top
这里也可以直接使用列表,使用如下:
self.stack[-1]
3.5 清空栈#
# 清空栈
def empty(self):
self.stack = []
self.top = -1
3.6 判断是否为空栈#
# 判断是否为空栈
def isEmpty(self):
if self.top == -1:
return True
else:
return False
这里可以直接判断列表是否为空即可,使用如下:
return self.stack is []
3.7 判断是否为满栈#
# 是否为满栈
def isFull(self):
if self.top == self.size - 1:
return True
else:
return False
3.8 完整代码#
#!usr/bin/env python
# encoding:utf-8
class Stack:
def __init__(self):
self.stack = [] # 栈列表
self.size = 20 # 栈大小
self.top = -1 # 栈顶位置
# 元素进栈
def push(self, element):
if self.isFull(): # 如果栈满,引发异常
raise StackException('Stack is full')
else:
self.stack.append(element)
self.top += 1
# 元素出栈
def pop(self):
if self.isEmpty(): # 如果栈为空,则引发异常
raise StackException('Stack is empty')
else:
element = self.stack[-1]
self.top -= 1
del self.stack[-1]
return element
# 获取栈顶位置
def getTop(self):
return self.top
# 清空栈
def empty(self):
self.stack = []
self.top = -1
# 判断是否为空栈
def isEmpty(self):
if self.top == -1:
return True
else:
return False
# 是否为满栈
def isFull(self):
if self.top == self.size - 1:
return True
else:
return False
class StackException(Exception): # 自定义异常类
def __init__(self, data):
self.data = data
def __str__(self):
return self.data
'''
主函数
'''
if __name__ == '__main__':
stack = Stack() # 创建栈
for i in range(10):
stack.push(i) # 元素进栈
print('栈顶位置:', stack.getTop())
while not stack.isEmpty():
print(stack.pop()) # 元素出栈
stack.empty() # 清空栈
for i in range(21):
stack.push(i) # 引发异常