包含min函数的栈

2020-05-07  本文已影响0人  su945

题目描述

定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
注意:保证测试中不会当栈为空的时候,对栈调用pop()或者min()或者top()方法。

问题分析

利用两个栈,分别存储最小值和正常数据。

解题思路1

class Solution {
public:
    void push(int value) {

        sData.push(value);
        if (sMin.empty())
        {
            sMin.push(value);
        }
        else
        {
            if (sMin.top() > value)
            {
                sMin.push(value);
            }
        }

    }
    void pop() {
        if (sData.top() == sMin.top())
        {
            sMin.pop();
        }
        else
        {
            sData.pop();
        }

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

    }
    int min() {
        
        return sMin.top();
    }
private:
    stack<int> sData;
    stack<int> sMin;
};
上一篇下一篇

猜你喜欢

热点阅读