包含min函数的栈 ---- 利用栈

题目:定义栈的数据结构,要求添加一个min函数,能够得到栈的最小元素。要求函数min、push以及pop的时间复杂度都是O(1)。

分析:google的一道面试题。我看到这道题目时,第一反应就是每次push一个新元素时,将栈里所有逆序元素排序。这样栈顶元素将是最小元素。但由于不能保证最后push进栈的元素最先出栈,这种思路设计的数据结构已经不是一个栈了。

在栈里添加一个成员变量存放最小元素(或最小元素的位置)。每次push一个新元素进栈的时候,如果该元素比当前的最小元素还要小,则更新最小元素。

乍一看这样思路挺好的。但仔细一想,该思路存在一个重要的问题:如果当前最小元素被pop出去,如何才能得到下一个最小元素?

注意读题:对空间复杂的没有特别的要求,完全可以借助辅助空间!

因此仅仅只添加一个成员变量存放最小元素(或最小元素的位置)是不够的。我们需要一个辅助栈。每次push一个新元素的时候,同时将最小元素(或最小元素的位置。考虑到栈元素的类型可能是复杂的数据结构,用最小元素的位置将能减少空间消耗)push到辅助栈中;每次pop一个元素出栈的时候,同时pop辅助栈。


代码如下:

#include<iostream>
#include<stack>

using namespace std;

class MinStack
{
public:
	void Push(int data);
	void Pop();
	int Min();
	void Print();//自定义一个输出栈元素的函数
private:
	stack<int> s;
	stack<int> s_GetMin;//辅助栈
};

void MinStack::Push(int data)
{
	if (s.empty())
		s_GetMin.push(data);
	else
	{
		int min = Min();
		if (data < min)
			s_GetMin.push(data);
		else
			s_GetMin.push(min);
	}

	s.push(data);
}

void MinStack::Pop()
{
	s_GetMin.pop();
	s.pop();
}

int MinStack::Min()//求得栈s此时的最小值
{
	return s_GetMin.top();
}

void MinStack::Print()
{
	cout << "分别输出栈s和辅助栈s_GetMin的元素(左竖是s,右竖是s_GetMin,数据从上往下的方向是从栈顶往栈底):\n";

	while (!s.empty())
	{
		cout << s.top() << " ";
		s.pop();

		cout << s_GetMin.top() << endl;
		s_GetMin.pop();
	}

	cout << endl;

}
int main()
{
	MinStack minStack;

	minStack.Push(4);
	minStack.Push(1);
	minStack.Push(2);
	minStack.Push(1);
	minStack.Push(3);
	minStack.Push(5);

	minStack.Print();

	return 0;
}

数据测试:








题目来自: http://www.acmerblog.com/stack-with-min-5990.html


  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值