实现stack 加上·getMin功能时间复杂度为O(n)

最新推荐文章于 2022-07-09 09:00:00 发布

djiomldddior199403

最新推荐文章于 2022-07-09 09:00:00 发布

阅读量106

点赞数

文章标签： java

原文链接：http://www.cnblogs.com/caobojia/p/6763702.html

版权

package com.hzins.suanfa;

import java.util.Stack;
/**
 * 实现stack 加上·getMin功能 时间复杂度为O(n)
 * @author Administrator
 *
 */
public class GetMinStack {
    private Stack<Integer> stackData;
    private Stack<Integer> stackMin;
    public GetMinStack(){
        this.stackData = new Stack<Integer>();
        this.stackMin = new Stack<Integer>();
    }
    public void push(int newNum){
        if(this.stackMin.isEmpty()){
            stackMin.push(newNum);
        }else if(newNum <this.getMin()){
            stackMin.push(newNum);
        }else{
            stackMin.push(this.getMin());
        }
        stackData.push(newNum);
    }
    public int pop(){
        if(stackData.isEmpty()){
            throw new RuntimeException("Your stack is empty");
        }
        stackMin.pop();
        return stackData.pop();
    }
    public int getMin(){
        if(this.stackMin.isEmpty()){
            throw new RuntimeException("Your stack is empty");
        }
        return this.stackMin.peek();
    }
}