今天面试被问到了实现一个含有max方法的栈。
本来还很庆幸这个题简单,用一个辅助栈存储最大值就可以了。没想到面试官要求只能使用常数内存,想了半天也没给出他期望的答案,回来搜一搜网上才发现O(1)空间复杂度的算法。
基本思想是维护一个最大值max。
在push方法中,比较当前的待压元素与max的差,若带压元素大于max,则更新max,并把这个差压入栈中,若小于max,直接压入这个差值(为负值)
在pop方法中,若栈顶是负数,则弹出的值为栈顶元素与max的和;若栈顶是正数,则弹出的值为max,并更新max为原max-栈顶元素。
下面来看个例子:
0, 2, 1, 4, 3 的入栈序列,栈中元素从底到顶依次为0, 2, -1, 2, -1;而max的变化依次为0, 2, 2, 4, 4。
退栈时弹出序列依次为4+(-1), 4, 2+(-1), 2, 0;max的变化依次为4, 4, 2, 2, 0。
含有min方法的栈也是一样的思路,就不再赘述了。