剑指offer 面试题30 包含min函数的栈

本文介绍了一种在栈数据结构中实现O(1)时间复杂度获取最小元素的方法,通过使用辅助栈存储最小值及次小值,确保在元素弹出后仍能快速找到新的最小值。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。

可能很多人一开始只想到用一个变量来记录最小值,每次更新最小值,但其实是不对的,因为考虑到有元素弹出的情况,当最小值弹出后,如何保持O(1)时间找到下一个最小值呢?

正确的解法:用一个辅助栈来存最小值和一系列次小值,栈顶存着最小值,当最小值弹出后,还会有次小值浮到栈顶。当弹出和删除时,对应地更新这个辅助栈。

#include <iostream>
#include <string>
#include <memory>
#include <vector>
#include<cmath>
#include<algorithm>
#include<stack>
using namespace std;

class Solution {
private:
   stack<int> st_x;
   stack<int> st_min;
public:
    void push(int value) {
        st_x.push(value);
        if(st_min.size()==0||value<st_min.top()) {
           st_min.push(value);
        }
    }
    void pop() {
        int pop_num=st_x.top();
        if(pop_num==st_min.top()) {
           st_min.pop();
        }
        st_x.pop();
    }
    int top() {
        return st_x.top();
    }
    int min() {
        return st_min.top();
    }
};

int main()
{
    Solution sol;
    sol.push(1);
    sol.push(2);
    cout<<sol.min()<<endl;
    system("pause");
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值