力扣,
思路不变,通过栈实现栈
对比学习:两个栈实现队列
这个题目相对更简单些
Python
class MinStack:
def __init__(self):
"""
initialize your data structure here.
"""
self.stack1 = list()
self.stack2 = list()
def push(self, x: int) -> None:
self.stack1.append(x)
if len(self.stack2) == 0:
self.stack2.append(x)
else:
tmp_min = min(self.stack2[-1], x)
self.stack2.append(tmp_min)
def pop(self) -> None:
self.stack1.pop()
self.stack2.pop()
def top(self) -> int:
return self.stack1[-1]
def getMin(self) -> int:
return self.stack2[-1]
# Your MinStack object will be instantiated and called as such:
# obj = MinStack()
# obj.push(x)
# obj.pop()
# param_3 = obj.top()
# param_4 = obj.getMin()
Java
class MinStack {
public Stack<Integer> s1;
public Stack<Integer> s2;
/** initialize your data structure here. */
public MinStack() {
s1 = new Stack<>();
s2 = new Stack<>();
}
public void push(int x) {
s1.push(x);
if (s2.isEmpty()) {
s2.push(x);
} else {
int min = Math.min(s2.peek(), x);
s2.push(min);
}
}
public void pop() {
s1.pop();
s2.pop();
}
public int top() {
return s1.peek();
}
public int getMin() {
return s2.peek();
}
}
Solution1:
辅助栈!
逻辑上要想清楚。。但是用栈结构来实现栈,目的不知为何。。。
class Solution {
public:
void push(int value) {
data.push(value);
if(data_min.empty()){ //辅助栈为空,则value直接压入辅助栈
data_min.push(value);
}
else { //辅助栈非空,
int stack_min = std::min(data_min.top(),value);
data_min.push(stack_min);
}
return;
}
void pop() {
data.pop();
data_min.pop();
return;
}
int top() {
return data.top();
}
int min() {
return data_min.top();
}
private:
stack<int> data;
stack<int> data_min;
};