LintCode 12.带最小操作的栈

第一种方法,用tree树排序,每次都排序

import java.util.LinkedList;
import java.util.TreeSet;

public class MinStack1 {
    /**
     * 带最小值操作的栈
     * 实现一个带有取最小值min方法的栈,min方法将返回当前栈中的最小值。
     * <p>
     * 你实现的栈将支持push,pop 和 min 操作,所有操作要求都在O(1)时间内完成。
     * <p>
     * 样例
     * 如下操作:push(1),pop(),push(2),push(3),min(), push(1),min() 返回 1,2,1
     * <p>
     * 注意事项
     * 如果堆栈中没有数字则不能进行min方法的调用
     *
     * 使用tree树进行排序,有点稀烂
     */

    protected LinkedList<Integer> linkedList;

    public MinStack1() {
        // do intialization if necessary
        linkedList = new LinkedList();
    }

    /*
     * @param number: An integer
     * @return: nothing
     */
    public void push(int number) {
        // write your code here
        linkedList.push(number);
    }

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

    /*
     * @return: An integer
     */
    public int min() {
        // write your code here
        TreeSet<Integer> treeSet = new TreeSet(linkedList);
        return treeSet.first();
    }
}

第二种方法,用两个stack一个记录每次push后的最小值,但是我感觉这样写是错误的,而且我觉得这道题的测试程序有问题

import org.junit.Test;

import java.util.LinkedList;

public class MinStack2 {
    /**
     * 带最小值操作的栈
     * 实现一个带有取最小值min方法的栈,min方法将返回当前栈中的最小值。
     * <p>
     * 你实现的栈将支持push,pop 和 min 操作,所有操作要求都在O(1)时间内完成。
     * <p>
     * 样例
     * 如下操作:push(1),pop(),push(2),push(3),min(), push(1),min() 返回 1,2,1
     * <p>
     * 注意事项
     * 如果堆栈中没有数字则不能进行min方法的调用
     * <p>
     * 使用tree树进行排序,有点稀烂
     */

//    采用两个栈储存,第二个栈压入最小元素。
//
//    已有小元素也压入mins栈,保持mins栈与date栈数量一样,原因是可以解决很多问题。若只用一个普通变量保存最小,若栈里有多个最小值,弹出不知道数量。
//
//    例如date中有[1,1,1,2,2,3,1] pop掉1,无法判断是否还有最小值。保持mins与date数量一样,没有跟新之前一直压入当前最小值
    private LinkedList<Integer> linkedList;
    private LinkedList<Integer> minStack;

    public MinStack2() {
        // do intialization if necessary
        linkedList = new LinkedList();
        minStack = new LinkedList<>();
    }

    /*
     * @param number: An integer
     * @return: nothing
     */
    public void push(int number) {
        // write your code here
        if (linkedList.isEmpty()) {
            minStack.offer(number);
        } else if (number <= minStack.peek()) {
//            minStack.clear();
            minStack.offerFirst(number);
        } else if (number == linkedList.peek()) {
            minStack.offer(number);
        }
        linkedList.push(number);
    }

    /*
     * @return: An integer
     */
    public int pop() {
        // write your code here
        if (linkedList != null) {
            int stackPop = linkedList.pop();
            if (stackPop == minStack.peek()) {
                minStack.pop();
            }
            return stackPop;
        } else {
            return 0;
        }
    }


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

    @Test
    public void testMinStack2() {
        MinStack2 minStack2 = new MinStack2();

        minStack2.push(-100);
        minStack2.min();
        minStack2.push(-99);
        minStack2.min();

    }
}
posted @ 2018-08-08 01:19  wei1  阅读(85)  评论(0编辑  收藏  举报