包含min函数的栈

基础知识:

1.比较两个数的大小min和max

c++ 使用 algorithm 提供的 max/min,返回最小最大的数。

#include<algorithm>

std::max(2, 3);
std::min(2, 3);
int temp = stack.top();
std::min(temp,2);

2.题目:包含min函数的栈

本题难点: 通过建立辅助栈实现
数据栈 AA : 栈 AA 用于存储所有元素,保证入栈 push() 函数、出栈 pop() 函数、获取栈顶 top() 函数的正常逻辑。
辅助栈 BB : 栈 BB 中存储栈 AA 中所有 非严格降序 的元素,则栈 AA 中的最小元素始终对应栈 BB 的栈顶元素即 min() 函数只需返回栈 BB 的栈顶元素即可。
因此,只需设法维护好 栈 BB 的元素,使其保持非严格降序,即可实现 min() 函数的 O(1)O(1) 复杂度。


函数设计:
push(x) 函数: 重点为保持栈 BB 的元素是 非严格降序 的。

将 xx 压入栈 AA (即 A.add(x) );
若 ① 栈 BB 为空 或 ② xx 小于等于 栈 BB 的栈顶元素,则将 xx 压入栈 BB (即 B.add(x) )。
pop() 函数: 重点为保持栈 A, B 的 元素一致性 。

0与-2对比-2较小,所以再赋值一个-2给min_stack,当执行pop时,是将两个栈一起各弹出,这样的好处是再添加数的时候不用判断谁小就加,不小就不加,而是只要小就加,而且再pop时也不用判断只有当两个栈顶数相等才弹出栈B。


top() 函数: 直接返回栈 AA 的栈顶元素即可,即返回 A.peek() 。

min() 函数: 直接返回栈 BB 的栈顶元素即可,即返回 B.peek() 。

#include<iostream>
#include<stack>
#include<algorithm>   //比大小头文件

class MinStack {
    std::stack<int> x_stack;
    std::stack<int> min_stack;     //辅助栈专门用来存放最小的数
public:
    MinStack() {
    }

    void push(int x) {
        x_stack.push(x);
        if (min_stack.empty())    //栈是空的就直接添加,不用对比
        {
            min_stack.push(x);
        }
        else 
        {
            int temp = min_stack.top();
            min_stack.push(std::min(temp, x));
        }
    }

    void pop() {
            x_stack.pop();       //两个一起弹出
            min_stack.pop();
    }

    int top() {
        return x_stack.top();
    }

    int min() {
        return min_stack.top();
    }
};

int main() {
    MinStack a;
    a.push(-2);
    a.push(0);
    a.push(-3);
    a.push(1);
    std::cout << a.min() << std::endl;
    a.pop();
    a.pop();
    std::cout << a.min() << std::endl;
}

运行结果

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值