博客主页 😑
Python数据结构(三)·栈

Author:

公爵

©

Wordage:

共计 2821 字

needs:

约 1 分钟

Popular:

9 ℃

Created:

目 录
栈是一种数据结构,只能从一端插入和删除操作,遵循着先进后出原则存储数据。

栈

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)  # 引发异常
共计 0 条评论,点此发表评论
萌ICP备20226257号 赣ICP备2022001242号-1 闽公网安备35020502000606号 本站已运行 253 天 9 小时 38 分 51统计 百度统计 本站由提供CDN加速服务 自豪地使用 Typecho 建站,并搭配 MyDiary 主题 Copyright © 2022 ~ 2022. 公爵书房 All rights reserved.
历史足迹
分类目录
  • 随笔
  • 知识
  • 分享
  • 书籍
  • 专栏
  • 笔记