数据结构——python 栈

所有动画演示可以在VisuAlgo这个网站上找到。

栈

栈（stack）又名堆栈，它是一种运算受限的线性表。其限制是仅允许在表的一端进行插入和删除运算。

栈允许进行插入和删除操作的一端称为栈顶(top)，另一端为栈底(bottom)；栈底固定，而栈顶浮动；栈中元素个数为零时称为空栈。插入一般称为进栈（PUSH），删除则称为退栈（POP）。

复杂度分析
栈属于常见的一种线性结构，对于进栈和退栈而言，时间复杂度都为 O(1)。

class Stack(object):
    def __init__(self, limit=10):
        self.stack = []  # 存放元素
        self.limit = limit  # 栈容量极限

    def push(self, data):  # push 进栈，将新元素放在栈顶
        # 判断栈是否溢出
        if len(self.stack) >= self.limit:
            raise IndexError('超出栈容量极限') # 抛出异常
        self.stack.append(data)

    def pop(self): # pop 退栈，从栈顶移出一个数据
        if self.stack:
            return self.stack.pop()  # 列表类的内置函数
        else:
            # 空栈不能被弹出元素
            raise IndexError('pop from an empty stack')

    def peek(self): 
        # 查看栈的栈顶元素（最上面的元素）
        if self.stack:
            return self.stack[-1]

    def is_empty(self):
        # 判断栈是否为空
        return not bool(self.stack) # not是一个布尔运算符，它返回值的布尔逆

    def size(self):
        # 返回栈的大小
        return len(self.stack)

栈的应用
在这个实验中，我们要求使用一个堆栈检查括号字符串是否平衡。

举例：
((())): True
((()): False
(())): False

def balanced_parentheses(parentheses):
    stack = Stack(len(parentheses))
    for parenthesis in parentheses:
        if parenthesis == '(':
            stack.push(parenthesis)
        elif parenthesis == ')':
            if stack.is_empty(): # 如果这时栈内为空，说明括号没有匹配完全
                return False
            stack.pop()
    return stack.is_empty() # 判断栈是否为空，如果为空，说明括号全部匹配
    
if __name__ == '__main__':
    examples = ['((()))', '((())', '(()))']
    print('Balanced parentheses demonstration:\n')
    for example in examples:
        print(example + ': ' + str(balanced_parentheses(example)))