码迷,mamicode.com
首页 > 其他好文 > 详细

Min Stack

时间:2018-12-09 10:46:29      阅读:156      评论:0      收藏:0      [点我收藏+]

标签:and   amp   ini   最小   port   empty   get   return   .com   

  

Design a stack that supports push, pop, top, and retrieving the minimum element in constant time.

  • push(x) -- Push element x onto stack.
  • pop() -- Removes the element on top of the stack.
  • top() -- Get the top element.
  • getMin() -- Retrieve the minimum element in the stack.

 

Example:

MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin();   --> Returns -3.
minStack.pop();
minStack.top();      --> Returns 0.
minStack.getMin();   --> Returns -2.

解题
使用2个栈s1,s2:
s1负责正常栈的储存。
s2储存每次出现最小的数的。
这样当s1删除到最小的数的时候,s2也删除栈顶的数。这样可以保证s2的栈顶一直是当前最小数。因为即使s1里面包含的当前第二小的数而s2不包含,需要用到这个第二小的数的时候,这个数早就被删除了。比如:
s1: 栈顶(2->1->3->4);
s2: 栈顶(1->3->4);
当删除1时,2早就不在s1中,所以接下来最小数是3。
 
class MinStack {
    
    Stack<Integer> element = new Stack<Integer>();
    Stack<Integer> minStack = new Stack<Integer>();
    public void push(int x) 
    {
        element.add(x);
        if (minStack.isEmpty() || x <= minStack.peek()) 
        {
           minStack.add(x);
        }
    }

    public void pop() 
    {
        if(element.isEmpty()) return;
        if (element.peek().equals(minStack.peek())) 
        {
             minStack.pop();
        }
        element.pop();
    }

    public int top() 
    {
        return element.peek();
        
    }

    public int getMin() 
    {
        return minStack.peek();
        
    }
}

 


reference:
作者:黑山老水
链接:https://www.jianshu.com/p/67f2c14accda
來源:简书
简书著作权归作者所有,任何形式的转载都请联系作者获得授权并注明出处。

Min Stack

标签:and   amp   ini   最小   port   empty   get   return   .com   

原文地址:https://www.cnblogs.com/hygeia/p/10090163.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!