/*******************************************************************
Copyright(c) 2016, Tyrone Li
All rights reserved.
*******************************************************************/
// 作者:TyroneLi
//
/*
Q:
包含min函数的矩阵:
定义栈的数据结构,请在该类型中实现一个可以得到栈的最小
元素的min函数,在该栈中,调用min,pop,push的时间复杂度都是O(1).
S:
定义一个辅助栈,原栈中每压入一个元素,都在辅助栈中压入新的最小在原栈中的最小值元素。
*/
#include <iostream>
#include <stack>
#include <cstdio>
#include <cassert>
#include <cstdlib>
#include <cstring>
template<typename T>
class StackWithMin
{
public:
StackWithMin() {}
~StackWithMin() {}
T&top();
const T&top() const;
void push(const T&value);
void pop();
const T&min() const;
bool empty() const;
size_t size() const;
private:
std::stack<T> m_data;
std::stack<T> m_min;
};
template<typename T>
T&StackWithMin<T>::top()
{
assert(m_data.size() > 0);
return m_data.top();
}
template<typename T>
const T&StackWithMin<T>::top() const
{
assert(m_data.size() > 0);
return m_data.top();
}
template<typename T>
void StackWithMin<T>::push(const T&value)
{
// 把新元素先添加到原栈
m_data.push(value);
// 当新元素比之前的最小元素小时,把新元素插入辅助栈里;
// 否则把之前的最小元素重复插入辅助栈里
if(m_min.size() == 0 || m_min.top() > value)
m_min.push(value);
else
m_min.push(m_min.top());
}
template<typename T>
void StackWithMin<T>::pop()
{
assert(m_data.size() > 0 && m_min.size() > 0);
m_data.pop();
m_min.pop();
}
template<typename T>
const T&StackWithMin<T>::min() const
{
assert(m_min.size() > 0);
return m_min.top();
}
template <typename T>
bool StackWithMin<T>::empty() const
{
return m_data.empty();
}
template <typename T>
size_t StackWithMin<T>::size() const
{
return m_data.size();
}
void Test(const char* testName, const StackWithMin<int>& stack, int expected)
{
if(testName != nullptr)
printf("%s begins: ", testName);
if(stack.min() == expected)
printf("Passed.\n");
else
printf("Failed.\n");
}
int main(int argc, char* argv[])
{
StackWithMin<int> stack;
stack.push(3);
Test("Test1", stack, 3);
stack.push(4);
Test("Test2", stack, 3);
stack.push(2);
Test("Test3", stack, 2);
stack.push(3);
Test("Test4", stack, 2);
stack.pop();
Test("Test5", stack, 2);
stack.pop();
Test("Test6", stack, 3);
stack.pop();
Test("Test7", stack, 3);
stack.push(0);
Test("Test8", stack, 0);
return 0;
}