包含min函数的栈

LeetCode 155. Min Stack
设计一个栈,支持如下操作,这些操作的算法复杂度需要是常数级,O(1)
1.push(x) : 将元素x压入栈中
2.pop() : 弹出(移除)栈顶元素
3.top() : 返回栈顶元素
4.getMin() : 返回栈内最小元素

class MinStack{
public:
    MinStack(){
    }//构造函数
    void push(int x){
    }//将元素x压入栈
    void pop(){
    }//将栈顶元素弹出
    int top(){
    }//返回栈顶元素
    int getMin(){
    }//返回站内最小元素
}

数据使用普通的栈data_stack存储,另外设置一个变量MIN,记录入栈过程中遇到的最小值,各项操作时有如下算法: 1.push(x) : 将元素x压入栈中,若x小于MIN,则更新变量MIN = x。
2.pop() : 弹出(移除)data_stack栈顶元素。
3.top() : 返回data_stack栈顶元素。
4.getMin() : 返回变量MIN。

分析

1.个变量MIN无法完成记录栈中所有状态的最小值,例如当栈进行pop操作的时候,数据栈更新了,也需要更新MIN变量的,但此时并未记录栈中第二小的元素,故没办法更新MIN变量。
2.栈的每个状态,都需要有一个变量记录最小值,每个状态即指无论对栈进行了push或pop操作, 该时刻的栈的最小值是被记录的。
3.在push或pop时,不能对数据进行排序,因为排序的复杂度不是O(1)。


算法设计

设置两个栈,数据栈data_stack与最小值栈min_stack,这两个栈对于添加元素push与弹出栈顶元素pop都是同步进行的:
1.push(x) : 将元素x直接压入数据栈data_stack中,若x小于最小值栈栈顶,则将x压入最小值栈中, 否则将最小值栈栈顶压入最小值栈。
2.pop() : 同时弹出(移除)数据栈栈顶与最小值栈顶元素。
3.top() : 返回数据栈data_stack栈顶元素。
4.getMin() : 返回最小值栈min_stack栈顶元素


class MinStack{
public:
    MinStack(){
}
    void push(int x){
        _data.push(x);//将数据压入数据栈
         if(_min.empty()){
             _min.push(x);
         }
        else{
            if(x>_min.top()){
                x = _min.top();
            }
            _min.push(x);
        }
}//比较当前数据与最小值栈栈顶数据大小,选择较小的压入最小值栈
void pop(){
    _data.pop();
    _min.pop();
}
int top(){
    return _data.top();
}
int getMin(){
    return _min.top();
}
private:
    std::stack<int> _data;//
    std::stack<int> _min;//
};
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 天气总让人那么手忙脚乱,酷热的许久实实考验着皮肤下黏膜的通透性,不觉间皮肤换了种新颜色,大地荡起了飞扬的灰粉。 老...
    b33a25aec52f阅读 286评论 1 1