LeetCode716.设计一个最大栈数据结构,既支持栈操作,又支持查找栈中最大元素。
实现MaxStack类:
示例:
对于peekMax(),我们可以用另一个栈来存储每个位置栈底的所有元素的最大值。例如,如果当前第一个栈中的元素为[2,1,5,3,9],那么第二个栈中的元素为[2,2,5,5,9]。在push(x)操作时,只需要将第二个栈的栈顶和xx的最大值入栈,而在pop()操作时,只需要将第二个栈进行出栈。
对于popMax(),由于我们知道当前栈中最大的元素值,因此可以直接将两个栈同时出栈,并存储第一个栈出栈的所有值。当某个时刻,第一个栈的出栈元素等于当前栈中最大的元素值时,就找到了最大的元素。因此我们将之前出第一个栈的所有元素重新入栈,并同步更新第二个栈,就完成了popMax()操作。