题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中,调用 min、push 及 pop 的时间复杂度都是 O(1)。
示例:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.min(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.min(); --> 返回 -2.
LeetCode :包含 min 函数的栈
解题思路
定义两个栈,dataStack 用于存储数据;minStack 用于存储数据栈中所含有数据的最小元素。 minStack 栈顶元素为当前栈中最小的值;minStack 中元素个数始终与 dataStack 的元素个数相同。
在对栈 dataStack 进行 push 入栈操作时,同时对 minStack 进行操作:当 minStack 中为空时直接压入栈中,当加入的元素比 minStack 中栈顶元素小时,将新加入的元素压入 minStack 中; 否则将 minStack 中的栈顶元素压入栈中。从而使 minStack 栈顶元素一直为当前栈中最小的值。
在对栈 dataStack 进行 pop 出栈操作时,同时对 minStack 进行出栈操作。
代码
class MinStack {
// 定义两个栈,dataStack 用于存储数据;minStack 用于存储数据栈中所含有数据的最小元素
private Stack