LintCode:12. 带最小值操作的栈

问题描述：
实现一个带有取最小值min方法的栈，min方法将返回当前栈中的最小值。

你实现的栈将支持push，pop 和 min 操作，所有操作要求都在O(1)时间内完成。

样例
如下操作：push(1)，pop()，push(2)，push(3)，min()， push(1)，min() 返回 1，2，1

注意事项
如果堆栈中没有数字则不能进行min方法的调用

在这个问题中，我使用的语言是java语言（其他语言忘光了写了估计都编译不了。。。），首先我们需要栈，恰好Java为我们提供了现成的栈Stack:
Java中色Stack继承的是Vector（应该很多人不知道的吧我也是上网翻资料找出来的）下面就介绍它的基本API

进栈

E push(E item) 
将项目推送到此堆栈的顶部。

出栈

E pop() 
删除此堆栈顶部的对象，并将该对象作为此函数的值返回。

判空

boolean empty() 
测试此堆栈是否为空。

取值而不出栈（正好取最小值用到）

E peek() 
查看此堆栈顶部的对象，而不从堆栈中删除它。

在代码实现中我们需要准备两个栈，一个正常进行压栈弹栈操作，另一个用来存储最小值，每次压栈前都与储存最小值的栈顶元素比较，确保最小值始终在栈顶。
下面贴出代码：（比较了一下，好像效率并不是很高介意勿看）

public class MinStack {
    //定义栈
    public Stack<Integer> stack;
    public Stack<Integer> min;
    public MinStack() {
            stack=new Stack();
            min=new Stack();
    }

    /*
     * @param number: An integer
     * @return: nothing
     */
    public void push(int number) {
        if(min.empty()||number<=min.peek()){
            min.push(number);
        }
        stack.push(number);
    }

    /*
     * @return: An integer
     */
    public int pop() {
        // write your code here
        int topstack=stack.pop();
        if(topstack==min.peek()){
            min.pop();
        }
        return topstack;

    }

    /*
     * @return: An integer
     */
    public int min() {
        // write your code here
        return min.peek();
    }
}

LintCode:12. 带最小值操作的栈

猜你喜欢