包含min函数的栈

---

title: 2019-8-17 包含min函数的栈

tags: 算法,每日一题,栈

---

包含min函数的栈

1. 问题描述

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

2. 题目解析

这里的很关键的一点是时间复杂度为O(1)。这说明肯定是要有个东西能够随时记录下当前栈中元素的最小值，并且在数据压入或者是弹出栈时也能够随时更新最小值。

2.1 解题思路

这道题有两种解题方法，但是本质其实是一样的：就是使用两个栈，一个栈用于存压入的数据，另一个栈用于存储当前栈的最小值，并且能够随着数据的压入和弹出动态的更新当前栈中的最小值。
第一种方式
使用两个栈，数据栈data_stack和最小元素栈min_stack，min_stack栈顶是当前数据栈对应的最小值。
入栈规则：对于要压入的数据value，先将这个数压入数据栈data_stack中。如过这个数value<=min_stack.top(),则将这个数压入min_stack中，否则压入min_stack.top()对应的值到min_stack。
出栈规则：从min_stack中弹出一个数据，从data_stack中弹出数据作为函数的返回结果。
min函数 ：返回min_stack.top()对应的值即可。

第二种方式
使用两个栈，数据栈data_stack和最小元素栈min_stack，min_stack栈顶是当前数据栈对应的最小值。
入栈规则：对于要压入的数据value，先将这个数压入数据栈data_stack中。如过这个数value<=min_stack.top(),则将这个数压入min_stack中。
出栈规则：从如果数据栈栈顶的数小于等于min_stack栈顶的数，则从min_stack中中弹出一个数据；从data_stack中弹出数据作为函数的返回结果。
min函数 ：返回min_stack.top()对应的值即可。

可以看出第二种方式要节省空间，这里只给出第二种方式的实现。

class Solution {
private:
    stack<int> data_stack;
    stack<int> min_stack;
public:
    void push(int value) {
        data_stack.push(value);
        if(min_stack.empty()) min_stack.push(value);
        if(value <= min_stack.top()) min_stack.push(value);
    }
    void pop() {
        if(min_stack.empty()) return;
        if(data_stack.top() == min_stack.top()){
            data_stack.pop();
            min_stack.pop();
            return;
        }
        data_stack.pop();
    }
    int top() {
        return data_stack.top();
    }
    int min() {
        return min_stack.top();
    }
};

更多关于编程和机器学习资料请关注FlyAI公众号。