剑指offer（20）——C++实现包含min函数的栈

题目

定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数（时间复杂度应为O（1））。

考察点

• 栈

解题思路

首先要保证它始终是一个栈的结构，首先要保证它始终是一个栈的结构，即弹出的始终是栈顶元素。那么这个时候就要使用辅助栈来完成存储和更新最小栈的作用，即每次压入的元素都是当前原来栈中的最小数。

完整代码

/*20-包含min函数的栈 */
#include<iostream>
#include<stack>
using namespace std;

class Solution {
public:
	void push(int value) {
		sdata.push(value);//将数据压入栈中

		//同时辅助栈中压入数据，并保持栈顶始终未最小值
		if (smin.empty() || value < smin.top())
			smin.push(value);
		else
			smin.push(smin.top());
	}
	void pop() {
		
		//if (smin.top() == sdata.top())
		smin.pop();
		sdata.pop();
	}
	int top() {
		return sdata.top();
	}
	int min() {
		return smin.top();
	}
private:
	stack<int> sdata;
	stack<int> smin;
};
int main()
{
	Solution s;
	
	//stack<int> data;
	for (int i = 0; i < 5; i++)
		s.push(i);
	cout << s.min() << endl;
	return 0;
}