剑指Offer19 包含min函数的栈

/*************************************************************************
> File Name: 19_MinInStack.cpp
> Author: Juntaran
> Mail: JuntaranMail@gmail.com
> Created Time: 2016年08月30日 星期二 19时29分48秒
************************************************************************/

#include <stdio.h>
#include <bits/stdc++.h>

using namespace std;

template<class T>
class minStack
{
private:
stack<T> stack_data;
stack<T> stack_min;

public:

void push(T &value)
{
// 新元素入栈
stack_data.push(value);

// 新元素比之前最小元素小的时候，新元素进入辅助栈
// 否则把之前最小元素重复插入辅助栈
if (stack_min.size()==0 || value<stack_min.top())
stack_min.push(value);
else
stack_min.push(stack_min.top());
}
T pop()
{
assert(stack_data.size()>0 && stack_min.size()>0);
stack_data.pop();
stack_min.pop();
return stack_data.top();
}
T min()
{
assert(stack_data.size()>0 && stack_min.size()>0);
return stack_min.top();
}
};

int main()
{
minStack<int> mStack;
for (int i = 0; i < 5; i++)
{
cout << i << " push" << endl;
mStack.push(i);
}
cout << "min is " << mStack.min() << endl;
for (int i = 1; i < 3; i++)
{
cout << mStack.pop() << " pop" << endl;
}
cout << "min is " << mStack.min() << endl;
for (int i = 5; i < 7; i++)
{
cout << i << " push" << endl;
mStack.push(i);
}
cout << "min is " << mStack.min() << endl;
for (int i = 1; i < 3; i++)
{
cout << mStack.pop() << " pop" << endl;
}
int k = -1;
mStack.push(k);
cout << "min is " << mStack.min() << endl;
cout << endl;
}