【面试题】剑指Offer-21-包含Min函数的栈

题目概述

这道题目很熟悉了

在栈和队列面试题中遇到并讲解过

解决方法

在栈中多用一个栈来保存最小值

当Min栈为空，或者入栈的元素小于Min栈的栈顶元素，则在入元素栈的同时

将该元素也一并入到Min栈中

从而保证了Min栈的栈顶元素永远都是最小值

出栈时，如果出栈的元素是最小元素，则也弹出Min栈的栈顶元素

否则，仅仅元素出栈即可

代码实现

#include<iostream>
using namespace std;
#include<stack>
#include<assert.h>
template<typename T>
class StackWithMin
{
public:
	//特点，存元素的栈和最小元素栈的元素大小是相同的
	//这里可以优化，如果小的话就存，否则最小元素栈再存
	//改动后，出栈时判断出栈的元素和最小元素栈的栈顶元素大小关系，如果相等，最小元素栈要出栈
	void Push(const T& data)
	{
		s_Elem.push(data);

		//判断入栈的元素和最小元素栈的栈顶元素的大小关系
		//来判断最小元素栈入的元素的值
		if (s_Min.empty() || data < s_Min.top())
			s_Min.push(data);
		else
			s_Min.push(s_Min.top());
	}

	void Pop()
	{
		assert(!s_Min.empty() && !s_Elem.empty());

		s_Min.pop();
		s_Elem.pop();
	}

	const T& Min()const
	{
		return s_Min.top();
	}
protected:
	stack<T> s_Elem;
	stack<T> s_Min;
};