描述: 定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数,并且调用 min函数、push函数 及 pop函数的时间复杂度都是 O(1) push(value):将value压入栈中 pop():弹出栈顶元素 top():获取栈顶元素 min():获取栈中最小元素
示例: 输入: [“PSH-1”,“PSH2”,“MIN”,“TOP”,“POP”,“PSH1”,“TOP”,“MIN”] 输出: -1,2,1,-1 解析: "PSH-1"表示将-1压入栈中,栈中元素为-1 "PSH2"表示将2压入栈中,栈中元素为2,-1 “MIN”表示获取此时栈中最小元素==>返回-1 "TOP"表示获取栈顶元素==>返回2 "POP"表示弹出栈顶元素,弹出2,栈中元素为-1 "PSH-1"表示将1压入栈中,栈中元素为1,-1 "TOP"表示获取栈顶元素==>返回1 “MIN”表示获取此时栈中最小元素==>返回-1 (题目来自牛客网)
用C++实现如下
class Solution {
//思路:定义两个stack,stack1用来存储所有的数据,stack2用来存储符合栈顶为最小数字的集合
private:
stack<int>s1;
stack<int>s2;
public:
void push(int value) {
s1.push(value); //push函数对于s1而言,所有的值都需要push进来
//对于s2而言,s2为空可以直接push进来;s2不为空,如果value值小于s2的顶部元素则push进来,否则再存一次s2的栈顶元素
if(s2.empty())
{
s2.push(value);
}
else
{
if(value < s2.top())
{
s2.push(value);
}else{
s2.push(s2.top());
}
}
}
//pop操作对于s1和s2而言,需要弹出;
void pop() {
s1.pop();
s2.pop();
}
//pop操作是对于s1而言
int top() {
return s1.top();
}
//min操作是对于s2而言
int min() {
return s2.top();
}
};
纯手撕代码,如果觉得内容不错麻烦点个赞,后面陆续配上Top100算法题通俗易懂的讲解视频,可以花两个月时间完全掌握,进大厂不是梦,转行狗亲测!
|