栈是一个集合,具有下面的2种基本操作
push: 把元素加入集合,这个过程我们叫做压入
pop: 把最后加入集合的元素从集合中移除,这个过程我们叫做推出
所以栈在移除元素的时候是遵循FILO( first in,last out),也就是先进后出的原则的。
使用python list来实现栈
我们主要是实现pop和push操作,另外再实现is_empty()方法来判断栈是否为空。
栈为空的意思是栈内没有任何元素了。
我们使用python的list来实现栈,这样比较简单。
import unittest
class Stack():
def __init__(self):
self.data = []
def __len__(self):
return len(self.data)
def push(self,ele):
self.data.append(ele)
def pop(self): # 默认出的是栈顶元素
return self.data.pop()
def is_empty(self):
return self.__len__() == 0
class StackTestCase(unittest.TestCase):
s = Stack() # 实例化一个栈对象
def test_1push(self):
self.s.push(1)
self.assertEqual(len(self.s),1)
def test_2pop(self):
self.assertEqual(self.s.pop(), 1)
self.assertEqual(len(self.s), 0)
self.assertTrue(self.s.is_empty())
if __name__ == '__main__':
unittest.main()
运行结果如下:
定义的类Stack,实现了栈的基本操作,如入栈,出栈,栈的长度,以及判断栈是否为空。
另外定义了StackTestCase类,继承了unitTest中的unittest.TestCase方法,接下来自定义两个函数test_1push()以及test_2pop()用来分别验证栈的基本的操作是否实现。
备注:可以看出,push()实际就是调用list.append()实现的,而pop()则是调用list.pop()来实现。