剑指offer——包含min的栈

题目描述:

定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。

这道题思路很简单,两个栈来实现,可是刷leetcode时遇到一种情况,使原来的答案出现了错误,所以在这里更新一下,下面是原来的代码:

代码一

出现一种情况,上面的代码会出现问题:

push(-10) push(14) push(-20) pop() push(10)

错误

   原因就在于 pop 后没有及时更新 mVal 的值造成的,所以可以在上面基础上进行修改。

修改pop函数

这回就不会出现错误了。

你可能感兴趣的:(剑指offer——包含min的栈)