数据结构中的入栈出栈代码
时间: 2024-09-21 08:06:30 浏览: 36
数据结构中的栈是一种特殊的数据结构,遵循“先进后出”(Last In First Out, LIFO)的原则,类似于我们生活中叠盘子的情况。下面是使用Python语言实现基本栈操作的代码示例:
```python
class Stack:
def __init__(self):
self.stack = []
# 入栈(添加元素)
def push(self, item):
self.stack.append(item)
# 出栈(移除并返回顶部元素)
def pop(self):
if not self.is_empty():
return self.stack.pop()
else:
raise Exception("Stack is empty")
# 查看栈顶元素但不删除
def peek(self):
if not self.is_empty():
return self.stack[-1]
else:
return None
# 检查栈是否为空
def is_empty(self):
return len(self.stack) == 0
# 返回栈的大小(元素个数)
def size(self):
return len(self.stack)
# 使用示例
stack = Stack()
stack.push(1)
stack.push('two')
print(stack.peek()) # 输出 'two'
print(stack.pop()) # 输出 'two'
```
在这个例子中,`push()`方法用于入栈,`pop()`方法用于出栈,`peek()`查看栈顶元素但不删除,`is_empty()`检查是否为空。栈的基本操作就包括这些。
阅读全文