包含min函数的栈

时间限制:C/C++ 1秒,其他语言2秒 空间限制:C/C++ 32M,其他语言64M 

题目描述

定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
注意:保证测试中不会当栈为空的时候,对栈调用pop()或者min()或者top()方法。
 
思路:
   第一眼看到这个题目,真是一脸懵逼,无从下手
   题目的目的很明确要实现输出栈中的最小元素的函数,使用栈的数据结构,所以需要定义stack类型的变量,需要实现push函数,pop函数,top函数等
  
  设立两个StackDate和StackMin,一个就是普通的栈,另外一个存储push进来的最小值。(stackMin的作用相当于是用于检索stackData中最小元素位置,)
  push压栈:
  每次压入的数据newNum都push进StackDate中,然后判断StackMin是否为空,如果为空那也把newNum同步压入StackMin里;如果不为空,就先比newNum和StackMin中栈顶元素的大小,如果newNum较大,那就不压入StackMin里,否则就同步压入StackMin里。

  Pop出栈:

    和push操作相对应,判断stackData栈顶元素是否与stackMin栈顶元素相同(stackMin中元素 <= stackData元素),如果相等,两个栈都弹出栈顶元素,不过不相等就stackData弹出

  代码如下:

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

附上堆栈stack的基本操作:

 

 

 

 

posted @ 2020-03-20 22:20  牛犁heart  阅读(161)  评论(0编辑  收藏  举报