剑指offer21包含min函数的栈

最新推荐文章于 2022-03-18 21:51:16 发布

原创最新推荐文章于 2022-03-18 21:51:16 发布 · 207 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种特殊栈的设计方案，通过维护两个栈——数据栈和辅助栈，实现了在O(1)时间内获取栈内最小元素的功能。文章详细展示了如何在进行push、pop操作的同时更新辅助栈以确保快速获取最小值。

定义栈的数据结构，请在该类型中实现一个能够得到栈最小元素的min函数。

//一切问题自己定，只要能实现功能就行了
//要求min push pop的时间复杂度都为o（1）
//给你一个栈，你要能在o(1)时间找到最小值
class Solution {
public:

void push(int value) {
//在压入数据栈的时候，把最小值也同时压入辅助栈
dataStack.push(value);//往栈中压入数值
if(minStack.empty())//如果辅助栈为空，则辅助栈压入数据
minStack.push(value);
else if(!minStack.empty()&&minStack.top()>=value)//如果辅助栈不为空，新压入的数据小于原来栈顶的数据
{
minStack.push(value);//直接压入辅助栈
}
else//如果新压入的数据比下面的数据大
{
int min=minStack.top();//当前最小值就是栈的顶部
minStack.push(min);//把该值再次仿佛栈顶
}

}
void pop() {//删除栈最小元素
dataStack.pop();
minStack.pop();
}
int top() {//返回顶部元素
return minStack.top();//
}
int min() {//找出最小元素
return minStack.top();//关键是这一个
}
private:
stack<int>dataStack;//数据栈（实际栈）
stack<int>minStack;//辅助栈
};