您的位置:首页 > 其它

栈和队列——设计一个有getMin功能的栈(一)

2017-08-29 13:20 579 查看
【题目】

  设计一个特殊的栈,在实现栈的基本功能的基础上,再实现返回栈中最小元素的操作

  

【要求】

  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();
}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  栈和队列