剑指Offer:面试题21——包含min函数的栈(java实现)

问题描述:

定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的min函数。在该栈中,调用min,push及pop的时间复杂度都是O(1).

思路:加入一个辅助栈用来存储最小值集合

(这里要注意题目并没有说栈内的元素类型,因此要尽量通用)

import java.util.Stack;

public class Solution {


    Stack s1=new Stack();
    Stack min=new Stack();
    public void push(int node) {
        if(min.empty()){
            min.push(node);
        }else{
            int top=(int)min.peek();
            if(node<top){
               min.push(node);
            }else{
                min.push(top);
            }
        }
        s1.push(node);
    }

    public void pop() {
        if(!(s1.empty())){
            s1.pop();
            min.pop();
        }
    }

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

    public int min() {
       if(min.empty()){
           return 0;
       }
       return (int)min.peek();    
    }





}
posted @ 2016-07-07 10:58  一只猫的爱园  阅读(107)  评论(0编辑  收藏  举报