包含min函数的栈

题目描述

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

代码：

class Solution {
public:
    stack<int > dataStack,minStack;   //一个是数据栈，用来正常存储压入的数据，
      //一个是最小栈，用来存储当前压入数据后此时数据栈中的最小值，并都保存在栈顶
    void push(int value) {
        dataStack.push(value);//数据栈正常压入数据
        if(minStack.empty()) minStack.push(value); //当最小栈为空，即第一次入栈时，此时的值就认为是当前最小值
        if(!minStack.empty()) {  //当最小栈非空时
            int tmp = minStack.top();
            if(value <=tmp) minStack.push(value);  //比较当前压入数据与最小栈顶数据的大小，若满足则压入最小栈，否则不作处理
        }
    }
    void pop() { //这个函数是为了处理当数据栈中弹出栈顶数据时，此时对最小栈的处理。
        //由于最小栈要时刻保持当前栈顶数据始终为当前数据栈中数据的最小值。所以分2种情况处理
        if(dataStack.top()==minStack.top()){//数据栈顶与最小栈顶相等时，弹出前者的栈顶后，最小值会变大，
            //又最小栈中从栈顶到栈底数据依次变大（包含等于），所以数据栈弹出栈顶时需要同步弹出最小栈顶
            dataStack.pop();
            minStack.pop();
        }
        else 
           dataStack.pop();//若两者栈顶不相等，此时数据栈顶比最小栈顶大，弹出数据栈顶不影响最小值
    }
    int top() {  //取的是数据栈顶
       return dataStack.top();  
    }
    int min() {
        return minStack.top();//最小栈顶始终是当前压入数据的最小值
    }
};