包含min函数的栈

2 篇文章 0 订阅

题目:定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的min函数。在该栈中,调用min、push以及pop的时间复杂度都是o(1)。

分析:看到这个问题,我们第一反应是每次压入一个新元素进栈时,让栈中所有的元素排序,让最下的元素位于栈顶,这样就可以在O(1)时间得到最小元素。但是这种思路不能保证最后压入栈的元素最先出栈,因此这个数据结构已经不是栈了。

可以在栈内添加一个成员变量存放最小元素,但是如果最小元素被弹出了之后,就需要得到次小的元素,因此我们可以借助一个辅助栈,来存放最小元素。

测试用例:新压入栈的元素比之前的最小值大,新压入栈的元素比之前的最小值小,弹出栈的数字不是最小的元素,弹出栈的元素是最小的元素。

代码如下:

template <typename T>void StackWithMin<T>::push(const T&  value)
{
	m_data.push(value);
	if(m_min.size()==0 || value<m_min.top())
		m_min.push(value);
	else
		m_min.push(m_min.top());
}
template <typename T> void StackWithMin<T>::pop()
{
	assert(m_data.size()>0 && m_min.size()>0);
	m_data.pop();
	m_data.pop();
}
template <typename T> const int& StackWithMin<T>::min() const()
{
	assert(m_data.size()>0 && m_min.size()>0);
	return m_min.top();
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值