剑指offer刷题———包含min函数的栈

1.问题重述

题目:定义栈的数据结构，请在该类型中实现一个能够得到全栈最小元素的min函数，同时要求，调用min、push、pop的时间复杂度都为O(1);

2.思路解析

本题目要求在栈中有一个min函数能够得到全栈最小的元素，同时还要保证栈的FILO的特性，因此这个最小的元素肯定要保存在外部，要不然会破坏栈的FILO特性。

所以我们可以创建两个栈来实现题目要求的数据类型，一个栈_data用来存放数据，一个栈_min用来存放当前的最小元素。当_data中添加一个元素的时候，_min中就需要添加一个当前的最小元素。同时在_min中插入新元素的时候，只需要将新插入的元素与栈顶元素比较，因为栈顶元素是插入之前全栈最小的元素，如果新插入的元素比栈顶小，插入这个元素，如果新插入的元素比栈顶大，那么我们将栈顶元素再次插入。我们来看下图：

3.代码实现

#include<iostream>
#include<stack>
using namespace std;
template<typename T>
class StackWithMin
{
public:
	void Push(T &&value)
	{
		_data.push(value);
		if (_min.size() < 1 || value < _min.top())
			_min.push(value);
		else
			_min.push(std::forward<T>(_min.top()));
	}
	void Pop()
	{
		if (!_data.empty())//一个不为空，那么两个肯定不为空，两个栈的长度是一样的
		{
			_data.pop();
			 _min.pop();
		}
	}
	T& Min()
	{
		if (!_data.empty())
		{
			return _min.top();
		}
	}
private:
	stack<T> _data;//存储数据
	stack<T> _min;//存储最小元素
};
int main()
{
        StackWithMin<int> s;
	s.Push(3);
	cout << "当前最小为:" << s.Min() << endl;
	s.Push(4);
	cout << "当前最小为:" << s.Min() << endl;
	s.Push(2);
	cout << "当前最小为:" << s.Min() << endl;
	s.Push(1);
	cout << "当前最小为:" << s.Min() << endl;
	s.Pop();
	cout << "当前最小为:" << s.Min() << endl;
	s.Pop();
	cout << "当前最小为:" << s.Min() << endl;
	s.Push(0);
	cout << "当前最小为:" << s.Min() << endl;
	return 0;
}

打印结果为：