码迷,mamicode.com
首页 > 其他好文 > 详细

设计一个获取最小值时间复杂度为O(1)的栈

时间:2020-02-29 00:55:11      阅读:93      评论:0      收藏:0      [点我收藏+]

标签:元素   不可   源代码   分析   get   erro   出栈   基本   public   

【题目】

  • 实现一个栈,在实现栈的基本功能的前提下,再实现返回最小元素的操作。

【要求】

  • pop、push、getMin操作的时间复杂度都是O(1)
  • 设计的类可以使用现成的栈结构。

【分析】

  • 想要使得获取最小值的时间复杂度为O(1),最简单的方法就是提前将最小值记录下来,当我们需要获取时便可直接获取
  • 栈的特点就是先进后出。它的操作具有一定规律,这使得我们可以很好的记录最小值。
  • 很容易就能够想到“双栈”来实现

【实现】

  • 实现语言:C++
  • 源代码如下:
#include<iostream>
#include<stack> 
using namespace std;

#define Error -1

// 一个有获取最小值功能的特殊栈,获取最小值的时间复杂度为O(1) 
class GetMinStack
{
    private:
    stack<int> m_BasicStack;                 //主栈
    stack<int> m_MinNumStack;                //最小值栈
    

    public: 
    void push(int value)                     //入栈
    {
        if(m_BasicStack.empty())            //如果栈中无数据
            m_MinNumStack.push(value);        
        else if(value<=m_MinNumStack.top())  //如果入栈的值比当前的最小值还要小
            m_MinNumStack.push(value);
        
        m_BasicStack.push(value); 
    } 
    
    int pop()
    {
        if(m_MinNumStack.empty())
        {
            //不可出栈
            cout<<"您操作的栈为空"<<endl;
            return Error; 
        }
        
        int temp=m_BasicStack.top();
        m_BasicStack.pop();
        if(temp==m_MinNumStack.top())
            m_MinNumStack.pop();
        
        return temp;  
    }
    
    int getMin()
    {
        if(m_MinNumStack.empty())
        {
            //不可出栈
            cout<<"您操作的栈为空"<<endl;
            return Error;  
        }
        return m_MinNumStack.top();
    }
};

int main()
{
    GetMinStack test;
    test.push(5);
    test.push(8);
    test.push(4);
    test.push(10);
    
    cout<<"该栈最小的数为:"<<test.getMin()<<endl; 
}
  • 运行截图
    技术图片

设计一个获取最小值时间复杂度为O(1)的栈

标签:元素   不可   源代码   分析   get   erro   出栈   基本   public   

原文地址:https://www.cnblogs.com/ZhuSenlin/p/12381055.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!