首页 > 其他 > 详细

实现一个栈,在基本功能的基础上,可以返回栈中最小值

时间:2020-04-29 09:23:17      阅读:65      评论:0      收藏:0      [点我收藏+]

import java.util.Stack;

/**
* 实现一个栈,在基本功能的基础上,可以返回栈中最小值
*/
public class GetMinStack {

public static class MyStack {

public Stack<Integer> stackData;

public Stack<Integer> stackMin;

public MyStack() {
this.stackData = new Stack<>();
this.stackMin = new Stack<>();
}

public void push(Integer value) {
if (stackMin.isEmpty()) {
stackMin.push(value);
} else if (value <= getMin()) {
stackMin.push(value);
} else {
stackMin.push(getMin());
}
stackData.push(value);
}

public Integer pop() {
if (stackData.isEmpty()) {
System.out.println("the stack is empty");
return null;
}
stackMin.pop();
return stackData.pop();
}

public Integer getMin() {
if (stackMin.isEmpty()) {
System.out.println("the stack is empty");
return null;
}
return stackMin.peek();
}

}

public static class MyStack2 {

public Stack<Integer> stackData;

public Stack<Integer> stackMin;

public MyStack2() {
this.stackData = new Stack<>();
this.stackMin = new Stack<>();
}

public void push(Integer value) {
if (stackMin.isEmpty()) {
stackMin.push(value);
} else if (value <= getMin()) {
stackMin.push(value);
}
stackData.push(value);
}

public Integer pop() {
if (stackData.isEmpty()) {
System.out.println("the stack is empty");
return null;
}
if (stackData.peek() == getMin()) {
stackMin.pop();
}
return stackData.pop();
}

public Integer getMin() {
if (stackMin.isEmpty()) {
System.out.println("the stack is empty");
return null;
}
return stackMin.peek();
}

}

}

/* 如有错误,欢迎批评指正 */

实现一个栈,在基本功能的基础上,可以返回栈中最小值

原文:https://www.cnblogs.com/laydown/p/12799227.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!