栈是一种后进先出(LIFO)的数据结构,在设计算法时有十分广泛的应用。对栈的操作是通过将元素压入栈(放在栈顶),或是将栈顶元素弹出,都是对栈顶元素的操作。若要实现一个高效的栈,则要考虑到这两种操作的时间复杂度。
有两种数据结构可以作为实现栈的基础:顺序表和链表。在顺序表尾端插入或删除元素的时间复杂度与在链表头部插入和删除元素的时间复杂度都是O(1),因此可以考虑将顺序表尾部或链表头部作为栈顶。
1.基于顺序表的实现
class SStack(): #基于顺序表技术实现的栈类
def __init__(self): #用list对象_elems存储栈中元素
self._elems=[] #所有栈操作都映射到list操作
def is_empty(self):
return self._elems==[]
def top(self): #取得栈顶元素,不删除
if self.is_empty():
print('为空栈!')
return
return self._elems[-1]
def push(self,elem): #将元素压入栈
self._elems.append(elem)
def pop(self): #将栈顶元素弹出
if self.is_empty():
print('为空栈!')
return
return self._elems.pop()
if __name__ == '__main__':
st=SStack()
for i in range(1,9,2):
st.push(i)
while not st.is_empty():
print(st.pop())
测试结果为
7
5
3
1
2.基于链表的实现
#节点类
class LNode:
def __init__(self,elem,next_=None):
self.elem=elem
self.next=next_
class LStack(): #基于链表技术实现的栈类,用LNode作为节点
def __init__(self):
self._top=None
def is_empty(self):
return self._top is None
def top(self):
if self.is_empty():
print('为空栈!')
return
return self._top.elem
def push(self,elem):
p=LNode(elem,self._top)
self._top=p
def pop(self):
if self.is_empty():
print('为空栈!')
return
p=self._top
self._top=self._top.next
return p.elem
if __name__ == '__main__':
st=LStack()
for i in range(1,10,2):
st.push(i)
while not st.is_empty():
print(st.pop())
测试结果为:
9
7
5
3
1