描述
实现一个带有取最小值min方法的栈,min方法将返回当前栈中的最小值。
你实现的栈将支持push,pop 和 min 操作,所有操作要求都在O(1)时间内完成。
如果堆栈中没有数字则不能进行min方法的调用
您在真实的面试中是否遇到过这个题?
样例
如下操作:push(1),pop(),push(2),push(3),min(), push(1),min() 返回 1,2,1
思路:用两个栈分别来存放当前输入元素【栈data】,以及当前栈里最小的元素【栈mins】(注意每一次都要往mins里入一个数)
C++实现:
class MinStack {
public:
std::stack<int> date;
stack<int> mins;
MinStack() {
// do intialization if necessary
}
/*
* @param number: An integer
* @return: nothing
*/
void push(int number) {
// write your code here
date.push(number);
if(mins.size()==0 || mins.top()>=number)
{
mins.push(number);
}
else{
mins.push(mins.top());
}
}
/*
* @return: An integer
*/
int pop() {
// write your code here
if(mins.size()>0 && date.size()>0)
{
int t = date.top();
date.pop();
mins.pop();
return t;
}
else{
return 0;
}
}
/*
* @return: An integer
*/
int min() {
// write your code here
if(mins.size()>0 && date.size()>0)
{
return mins.top();
}
return 0;
}
};
Py3:
class MinStack(object):
def __init__(self):
# do some intialize if necessary
self.stack = []
def push(self, number):
# write yout code here
if len(self.stack) == 0:
self.stack.append([number, number])
else:
self.stack.append([number, min(number, self.stack[-1][1])])
def pop(self):
# pop and return the top item in stack
return self.stack.pop()[0]
def min(self):
# return the minimum number in stack
return self.stack[-1][1]