【题目】
设计一个特殊的栈,在实现栈的基本功能的基础上,再实现返回栈中最小元素的操作
【要求】
1、pop、push、getMin操作的时间复杂度都是O(1)
2、设计的栈类型可以使用现成的栈结构
【代码实现】
//实现一个特殊的栈,在实现栈的基本功能的基础上在实现返回栈中最小元素的操作
import java.util.*;
class MyStack1 {
private Stack<Integer> stackData;
private Stack<Integer> stackMin;
public MyStack1(){
this.stackData = new Stack<Integer>();
this.stackMin = new Stack<Integer>();
}
//入栈
public void push(int newNum){
if(this.stackMin.isEmpty()){
this.stackMin.push(newNum);
}else if(newNum<=this.getmin()){
this.stackMin.push(newNum);
}
this.stackData.push(newNum);
}
//出栈
public int pop(){
if(this.stackData.isEmpty()){
throw new RuntimeException("Your stack is Empty");
}
int value = this.stackData.pop();
if(value==this.getmin()){
this.stackMin.pop();
}
return value;
}
//取最小值
public int getmin(){
if(this.stackMin.isEmpty()){
throw new RuntimeException("Your stack is Empty");
}
return this.stackMin.peek();
}
}