实现一个栈,要求Push(入栈),Pop(出栈),Min(返回最小值的操作)的时间复杂度为O(1)

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/x_y_r129/article/details/50854931

问题描述:实现一个栈,要求Push(入栈),Pop(出栈),Min(返回最小值的操作)的时间复杂度为O(1)
分析问题:要记录从当前栈顶到栈底元素的最小值,很容易想到用一个变量,每push一个元素更新一次变量的值。那么问题来了,当执行pop操作时,上一次的最小值就找不到了。
解决问题:这里有两种方法解决这个问题
方法一
使用一个栈。元素x入栈时,执行一次push(x),再push(min),min表示当前栈顶到栈底元素最小值;元素出栈时,执行两次pop()
这里写图片描述
举个栗子,如果我们要入栈的元素序列是递增的(1,2,3……1000),那么每一次入栈都要push(1),操作冗余。
方法二
使用两个栈s1和s2,s2做为辅助栈(每次压入s2的都是s1的最小值)。元素x入栈时,将x和s2栈顶元素做比较,如果x小于等于s2.top(),将x分别push到s1和s2,否则x只push到s1
;元素出栈时,将s1栈顶元素和s2栈顶元素做比较,如果s1.top()等于s2.top(),s1和s2都执行pop操作,否则只执行s1.pop()
这里写图片描述

代码实现(方法二)

#include<iostream>
#include<stack>
using namespace std;
typedef int T;


class StackWithMin
{
public:
    StackWithMin()
    {}
    ~StackWithMin()
    {}
    void Push(T x)
    {
        s1.push(x);
        if (s2.empty() || x <= s2.top())
        {
            s2.push(x);
        }
    }
    void Pop()
    {
        if (s1.top() == s2.top())
        {
            s2.pop();
        }
        s1.pop();
    }
    T Min()
    {
        if (!s2.empty())
        {
            return s2.top();
        }
    }
    void Print()
    {
        while (!s1.empty())
        {
            cout << s1.top() << "->";
            s1.pop();
        }
        cout << endl;
        while (!s2.empty())
        {
            cout << s2.top() << "->";
            s2.pop();
        }
        cout << endl;
    }
private:
    stack<T> s1, s2;
};

void Test()
{
    StackWithMin  s;
    s.Push(4);
    s.Push(3);
    s.Push(5);
    s.Push(2);
    s.Push(1);
    s.Push(1);
    s.Print();
}

int main()
{
    Test();
    system("pause");
    return 0;
}
展开阅读全文

没有更多推荐了,返回首页