[剑指offer]包含min函数的栈(最小栈)

题目:

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

注意：保证测试中不会当栈为空的时候，对栈调用pop()或者min()或者top()方法。

---

思路:

使用两个栈 

一个栈stack  push任何值(来一个值就push进去)  另一个栈stack_little  push比原stack_little栈顶元素小的值

因为当stack pop之后最小值可能会有变化 

当stack pop的元素和小栈stack_little栈顶元素不同时不影响此时栈的最小值  所以只需pop原栈顶的值 不影响栈的最小值

但是 当stack栈顶元素与stack_little栈顶元素相同时 两个栈需要同时pop 否则栈的最小值就不对了 因为min()函数是返回小栈栈顶元素的

---

代码:

class Solution {
  private:
	stack<int> _stack;
	stack<int> _stack_little;
public:
	void push(int value) {
		if (_stack.empty()) {
			_stack.push(value);
		}
		if (_stack_little.empty()) {
			_stack_little.push(value);
		}
		else if (value <= _stack_little.top()) {
			_stack_little.push(value);
		}
			_stack.push(value);
	}
	void pop() {
		if (_stack.empty()) {
			return;
		}
		if (_stack.top() == _stack_little.top()) {
			_stack.pop();
			_stack_little.pop();
		}
		else {
			_stack.pop();
		}
	}
	int top() {
		if (!_stack.empty()) {
				return _stack.top();
		}
	}
	int min() {
		if (!_stack_little.empty()) {
			return _stack_little.top();
		}
	}
};