剑指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();    
    }





}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章