Hello,大家好,今天Val来给大家分享,关于如何实现一个栈,让它的push、pop、返回最小值min的时间复杂度为O(1),和如何编写这个代码的测试用例。
基本算法思想
1.实现主要内容:
创建一个MyStack类,类中用两个栈实现这个算法:
一个栈里面放的是所有的数据,另一个辅助栈存放最小值;
每次栈1插入一个数据,这个数据与栈2的栈顶相比较,如果这个数据小于或者栈2的栈顶,那就把这个数放入栈2;
栈2的栈顶永远是当前所有数据的最小值。
2.栈2数据插入规则:
stack2内存放的是当前栈1中的最小值;
每次往栈1中放数据都要和栈2栈顶数据比较,既然存的是最小值,是否要把跟栈2栈顶数据相等的数据也放入栈2中?
放,必须放,实现的这个栈本来就是为了方便,如果与栈顶数据相等,却没有放进栈2中,那么在删除栈1中的元素时,如果这个元素与栈2栈顶元素相等,可是这个最小值只有一个,栈1中元素删除了,栈2中的这个最小值什么时候删除,你怎么知道你删除的时候栈1中不会再有这个数据,你如何确定你所删除的时候,栈1中就只剩下1个这个数据了???
所以,当前插入节点等于栈2栈顶数据时,也要把它放入栈2中,这样删除一个数据时,判断它是否和栈2栈顶数据相等,相等的话,把栈2的栈顶也要删掉,用这种方法,删除数据才不会出错,栈2的栈顶元素永远是当前栈1中所有元素中最小的那个。
3.图示:
4.代码