时间限制:C/C++ 1秒,其他语言2秒 空间限制:C/C++ 32M,其他语言64M 热度指数:439567
本题知识点: 栈
算法知识视频讲解
题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
注意:保证测试中不会当栈为空的时候,对栈调用pop()或者min()或者top()方法。
解题思路:
正确的做法是借助一个辅助栈。他们之间有一种对应关系:辅助栈的栈顶元素,就是原栈所有元素的最小值。
对原栈和辅助栈的处理过程如下:
元素压入原栈的时候,如果辅助栈为空,或者元素 <= 辅助栈的栈顶元素,那么将元素也压入辅助栈
元素弹出原栈的时候,如果元素等于辅助栈的栈顶元素,辅助栈也弹出元素
这里的判断条件是元素 <= 辅助栈的栈顶元素而不是元素 < 辅助栈的栈顶元素,是为了应对重复元素。例如将 1、2、3、1 依次入栈,采用错误的判断条件,那么辅助栈里面只有 1。在原栈弹出 1 之后,辅助栈为空,就没法获得原栈元素的最小值。
代码
class Solution {
public:
stack<int> index;
stack <int > Min;
void push(int value) {
index.push(value);
if(Min.size() == 0 || value < Min.top())
{
Min.push(value);
}
else
{
Min.push(Min.top());
}
}
void pop() {
index.pop();
Min.pop();
}
int top() {
Min.top();
int x = index.top();
return x;
}
int min() {
return Min.top();
}
};