题目:
实现一个特殊的栈,在实现栈的基本功能的基础上,再实现返回栈中最小元素的操作。
要求:
1、pop、push、getMin操作的时间复杂度都是O(1)
2、设计的栈类型可以输用现成的栈结构
首先看下我写的,思路大概就是用两个栈,一个栈用于自定义栈的弹出,压入,辅助栈用于存放最小值,当压入第一个栈的时候,会判断辅助栈是否为空,如果为空也压入,如果不为空,就比较,如果小于等于辅助栈的第一个值,就压入。弹出操作,如果第一个栈弹出的数和辅助栈的第一个数相等,也弹出辅助栈的第一个数,getMin()获取辅助栈的顶部的数据就好。
代码如下:
package 设计一个有getMin功能的栈;
import java.util.Stack;
/**
* 实现一个特殊的栈,在实现栈的基本功能的基础上,再实现返回栈中最小元素的操作
* ====================
* 要求:
* 1.pop、push、getMin操作的时间复杂都是O(1)
* 2.设计的栈类型可以使用现成的栈结构
*
* @author dream
*
*/
public class MyStack1 {
private Stack<Integer> stackData;
private Stack<Integer> stackMin;
public MyStack1()
{
stackData = new Stack<Integer>();
stackMin = new Stack<Integer>();
}
/**
* 压入操作
* @param newNum
*/
public void push(int newNum)
{
if(stackMin.isEmpty())
{
stackMin.push(newNum);
}
stackData.push(newNum);
if(newNum <= stackMin.peek())
{
stackMin.push(newNum);
}
}
/**
* 弹出操作
* @return
*/
public int pop()
{
if(stackData == null)
{
throw new RuntimeException("Your Stack isEmpty");
}
int value = stackData.pop();
if(stackMin.peek() == value)
{
stackMin.pop();
}
return value;
}
public int getMin()
{
if(stackMin.isEmpty())
{
throw new RuntimeException("Your stack is Empty.");
}
return stackMin.peek();
}
public static void main(String[] args) {
MyStack1 stack1 = new MyStack1();
stack1.push(3);
stack1.push(2);
stack1.push(1);
int stack = stack1.pop();
int min = stack1.getMin();
System.out.println(stack + "" + min + "");
}
}
看下书上是如何写的:
import java.util.Stack;
public 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();
}
}
可以看出我写的不规范,pop操作应该是if-else的关系。
书上给出第二种设计方案:
其实差不多,就是在push操作的时候,如果stackData压入的数据大于等于stackMin的顶部元素的时候,StackMin重复压入顶部元素,这样,pop的时候,每次MinStack都弹出就可以了。
代码如下:
import java.util.Stack;
public class MyStack2 {
private Stack<Integer> stackData;
private Stack<Integer> stackMin;
public MyStack2()
{
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);
}
else {
int newMin = this.stackMin.peek();
this.stackMin.push(newMin);
}
this.stackData.push(newNum);
}
public int pop() {
if (this.stackData.isEmpty()) {
throw new RuntimeException("Your stack is Empty.");
}
this.stackMin.pop();
return this.stackData.pop();
}
public int getmin() {
if (this.stackMin.isEmpty()) {
throw new RuntimeException("Your stack is empty.");
}
return this.stackMin.peek();
}
}
代码在我的Github上有,欢迎下载查看。