设计一个栈，提供一个函数能够找出栈内最大的元素，时间复杂度为O(1)

这是一道淘米网的面试题，当时面试官直接拿出一张纸来让我写出实现方法。我思考了一会，就是关于时间复杂度这块，实在是没有办法，因为我想的都是要

遍历一遍这个栈，这样肯定超了，在面试完回来之后，我思考了一段时间，网上了解了一下，得出一个解题思路

想要找出最大的数，且时间复杂度为O(1)，那么就一定要知道这个最大的数在哪放着，那么我们就可以一次性的找到他。

利用栈的特性，我可以一次压2个数，一个数是当前要入栈的数，另外一个是当前栈中最大的数，这样一次压两个，然后出栈也是一次出两个，这样我就可以每次入栈

时，将要入栈的数，和已经入栈的那个数一块压入的最大的数进行对比，看那个大，这样就可以每次保证，我压入的数其中是真正要 压入的，另外一个就是栈中最大的，这样的话，我找出最大的数，只需要将栈顶元素的下一个元素出栈即可，也实现了O(1)的时间复杂度。

下面是代码

stack：

#include <iostream>
#include <assert.h>
using namespace std;


#define MAX_SIZE 50 //栈容量


class stack
{
public:
stack();
~stack();
int find_max(); //找出栈内最大的数
bool push(int); //入栈
int get_second(); //得到元素入栈前栈内最大的数
int pop();
private:
int top;
int *element;
int capacity; //栈的大小
};


stack::stack():capacity(MAX_SIZE/2),top(0)
{
element = new int[MAX_SIZE];
assert(element != NULL);
}
stack::~stack()
{
if(element != NULL)
{
delete []element;
}
}
bool stack::push(int a)
{
if((top+1) >= capacity)
{
cout<<"stack is over flow"<<endl;
return false;
}


int tmp = get_second(); //如果栈内还没有元素，则返回-1
if(-1 == tmp)
{
tmp = a;
}


if(a >= tmp) //要入栈的元素大于栈内最大的数
{
element[top++] = a;
element[top++] = a;
return true;
}
element[top++] = tmp;
element[top++] = a;
return true;
}


int stack::get_second()
{
int tmp1;
if(top-1 <= 0)
{
return -1;
}
tmp1 = element[top-2];
return tmp1;
}


int stack::pop()
{
int tmp1,tmp2;
if(top <= 0)
{
cout<<"stack is empty"<<endl;
return -1;
}
tmp1 = element[--top];
tmp2 = element[--top];
return tmp1;
}




int stack::find_max()
{
if(top != 0 && top != 1) //当栈内元素个数是0个或者一个的话那就没有过元素入栈。
{
return(element[top-2]); 
}
return -1;
} 

main：

#include "stack.h"


int main()
{
stack st;
st.push(3);
st.push(5);
st.push(2);


int ret = st.find_max();
cout<<ret<<endl;


return 0;


}