设计一个有getMin功能的栈

设计一个有getMin功能的栈

【题目】

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

【要求】

pop、push、getMin的操作时间复杂度都是O(1)

设计的栈类型可以使用现成的栈结构

【解决思路】

在设计上，我们使用两个栈来解决问题，使用空间换取时间复杂度上的最优解。一个栈用来存储我们当前栈中最小的元素；另外的一个栈用来实现我们的基本栈的操作，存储基本数据。

数据压栈规则

假设，当前要压入栈的数据为newNum，先将其压入到stackData.然后判断stackMin是否为空：

1.如果为空：则将newNum压入到minStack

2.如果不为空，则将newNum和minStack栈顶的数据进行比较，如果比它小，则把newNum压入到minStack中去

数据出栈规则

先从stackData中执行出栈操作，取出数据，然后将这个数据和当前最小数据进行比较，如果当前出栈的数据是当前最小数据，则在minStack也进行出栈即可。

【代码实现】

package study0224;

import java.util.Stack;

public class MyStrack1 {

private Stack<Integer> strackData;
private Stack<Integer> minData;

public MyStrack1() {
this.strackData = new Stack<Integer>();
this.minData = new Stack<Integer>();
}

public void push(Integer newNum) {
if (strackData.isEmpty()) {
strackData.push(newNum);
minData.push(newNum);
} else {
strackData.push(newNum);
if (this.getMin() > newNum) {
minData.push(newNum);
}
}
}

public int pop() {
int popNum = strackData.pop();
if(popNum == this.getMin()) {
minData.pop();
}
return popNum;
}

public int getMin() {
if (this.minData.isEmpty()) {
throw new RuntimeException("min stack is null");
}
return minData.peek();
}

}