欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

包含min函数的栈

程序员文章站 2024-01-29 12:46:40
...

1.题目

定义栈的数据结构,请在该类型中实现一个能够得到栈最小元素的min函数。在该栈中调用push、pop、top、min的时间复杂度都是o(1)。

思路

定义一个辅助栈,用于存放数据栈每次入栈时的最小元素,辅助栈的栈顶元素是数据栈内最小值。
举例:创建一个空数据栈和空辅助栈。
1.数据栈中压入3,辅助栈为空栈,最小值为3,辅助栈压入3
2.数据栈中压入4,4大于辅助栈的栈顶元素3,最小值为3,辅助栈压入3
3.数据栈中压入2,2小于辅助栈的栈顶元素3,最小值为2,辅助栈压入2
4.数据栈中压入1,1小于辅助栈的栈顶元素2,最小值为1,辅助栈压入1
5.数据栈弹出栈顶元素,辅助栈弹出栈顶元素,辅助栈弹出栈顶元素后的栈顶元素是当前数据栈的最小值
6.数据栈弹出栈顶元素,辅助栈弹出栈顶元素,辅助栈弹出栈顶元素后的栈顶元素是当前数据栈的最小值
注意:数据栈和辅助栈操作同步,即对数据栈插入时,同时辅助栈也插入;弹出数据栈栈顶元素时,同时弹出辅助栈栈顶元素。
包含min函数的栈

class Solution {
public:
    void push(int value) 
    {
        data.push(value);

        if(data.empty() || value<smin.top())
            smin.push(value);
        else
            smin.push(smin.top());
    }

    void pop() 
    {
        if(!data.empty() && !smin.empty())
        {
            data.pop();
            smin.pop();
        }
    }

    int top() 
    {
        return data.top();
    }

    int min() 
    {
        return smin.top();
    }
private:
    stack<int> data;
    stack<int> smin;
};