原题
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
Reference Answer
思路分析
思路:使用一个辅助栈保存最小值
# -*- coding:utf-8 -*-
class Solution:
def __init__(self):
self.stack = []
self.min_num =[]
def push(self, node):
# write code here
self.stack.append(node)
if self.min_num and self.min_num[-1] < node:
self.min_num.append(self.min_num[-1])
else:
self.min_num.append(node)
def pop(self):
# write code here
if self.stack:
self.min_num.pop()
return self.stack.pop()
return None
def min(self):
# write code here
return self.min_num[-1] if self.min_num else None
C++ Version:
class Solution {
stack<int> store;
stack<int> store_min;
public:
void push(int value) {
store.push(value);
if (store_min.empty()){
store_min.push(value);
}
else{
if (value <= store_min.top()){
store_min.push(value);
}
else{
store_min.push(store_min.top());
}
}
}
void pop() {
store.pop();
store_min.pop();
}
int top() {
return store.top();
}
int min() {
return store_min.top();
}
};
Note:
python 中 list.pop
与 list.pop()
是不同的,自己就是误用 pop()
为 pop
一直报错!