剑指offer-20 -- 包含min函数的栈 - C++

题目描述

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

题解

题目抽象:要求实现一个O(1)时间复杂度的返回最小值的栈。正常情况下,栈的push,pop操作都为O(1),
但是返回最小值,需要遍历整个栈,时间复杂度为O(n),所以这里需要空间换时间的思想。

方法:使用辅助栈

首先需要一个正常栈normal,用于栈的正常操作,然后需要一个辅助栈minval,专门用于获取最小值,具体操作如下。

  • push操作就如图片上操作。
  • pop操作直接对应弹出就好了。
  • top操作就返回normal的栈顶
  • min操作就返回minval的栈顶
class Solution {
public:
    stack<int> normal, minval;
    void push(int value) {
        normal.push(value);
        if(minval.empty())
            minval.push(value);
        else{
            if(value <= minval.top())
                minval.push(value);
            else
                minval.push(minval.top());
        }
    }
    void pop() {
        normal.pop();
        minval.pop();
    }
    int top() {
        return normal.top();
    }
    int min() {
        return minval.top();
        
    }
};

 

©️2020 CSDN 皮肤主题: 编程工作室 设计师:CSDN官方博客 返回首页