剑指offer 21：包含min函数的栈

Posted 2022-12-24 fancy-li

题目描述

定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数（时间复杂度应为O（1））。

解题思路

要求在O(1)时间内完成，由于栈后进先出的原则，不能出现破坏栈结构的事情。因而采用以空间换时间的思路，建立辅助栈，辅助栈与数据栈的出入栈操作同步进行，区别在于数据栈压入当前元素，辅助栈压入当前情况下的最小元素，于是，需要定义一个临时变量来记录上一时刻的最小元素，并与当前元素比较，记录当前情况下的最小元素。出栈时，两个栈同步操作。

 

C++代码实现：

class Solution 
public:
    stack<int> m_data,auxiliary;
    int minVal;
    void push(int value) 
        if(m_data.size()==0)
         minVal=value;   
        else
            minVal=minVal>value?value:minVal;
        
        m_data.push(value);
        auxiliary.push(minVal);
    
    void pop() 
        m_data.pop();
        auxiliary.pop();
    
    int top() 
        return m_data.top();
    
    int min() 
        return auxiliary.top();
    
;