包含min函数的栈 ----《剑指offer》面试题21

题目

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

思路

  用一个std::stack<std::pair<int,int>>表示这个栈， pair<int，int>中 first表示要插入的值，second表示当前栈中最小的元素。 这样每次栈顶元素的second的值就是栈本身最小的值。当有元素入栈时，将元素与栈最小值比较，取小的那个，再压入栈中。

代码

#include <iostream>
#include <stack>
#include <utility>
using  namespace std;

std::stack<std::pair<int,int>> mystack;

int min(void)
{
    return  mystack.top().second;
}

void pushValue(const int &val)
{
    int minmum = 0;
    if (mystack.empty())
        minmum = val;
    else
        minmum = std::min(min(),val);

    mystack.push(pair<int,int>(val,minmum));
}


int main()
{
    pushValue(4);
    pushValue(3);
    pushValue(6);
    pushValue(3);
    pushValue(4);
    pushValue(2);

    cout << "current min : " <<  min() << endl;
    mystack.pop();
    cout << "current min : " <<  min() << endl;
    mystack.pop();
    cout << "current min : " <<  min() << endl;
    mystack.pop();
    cout << "current min : " <<  min() << endl;
    mystack.pop();
    cout << "current min : " <<  min() << endl;
    mystack.pop();
    cout << "current min : " <<  min() << endl;

    return EXIT_SUCCESS;
}