剑指offer: 包含min函数的栈

题目描述:定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。

首先 栈的特征   先进后出。

因此我们只要定义一个min_stack的栈,每push一个元素都与min_stack.top()元素进行大小比较,如果比min_stack.top()小的话就push进min_stack,这样就可以保证min_stack这个栈的top元素始终为最小的元素。

class Solution 
{
private:
	stack<int> min_stack;
	stack<int> my_stack;
public:
	void push(int value)
	{
		my_stack.push(value);

		if (min_stack.size() == 0)
		{
			min_stack.push(value);
		}
		if (min_stack.top() > value)
		{
			min_stack.push(value);
		}
	}
	void pop()
	{
		if (my_stack.top() == min_stack.top())
		{
			my_stack.pop();
			min_stack.pop();
		}
		else
		{
			my_stack.pop();
		}
	}
	int top() 
	{
		return my_stack.top();
	}
	int min() 
	{
		return min_stack.top();
	}

};

猜你喜欢

转载自blog.csdn.net/Shile975/article/details/88874569
今日推荐