【劍指offer】面試題21:包含min函數的棧

class StackWithMin:
	def __init__(self):
		self.data = []
		self.min_data = []
		self.size = 0

	def push(self, val):
		self.data.append(val)

		if len(self.min_data) == 0 or \
				self.min_data[-1] > val:
			self.min_data.append(val)

		else: 
			self.min_data.append(self.min_data[-1])
		self.size += 1


	def pop(self):
		if self.size > 0:
			del self.data[-1]
			del self.min_data[-1]
			self.size -= 1

	def min(self):
		return self.min_data[-1]

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章