设计有一个getMin功能的栈
说明
本文来自左程云老师所著的《程序员面试代码指南》第一章
题目
实现一个特殊的栈,在实现栈的基本功能的基础上,再实现返回栈中最小元素的操作。
要求
1.pop、push、 getMin操作的时间复杂度都是0(1)。
2.设计的栈类型可以使用现成的栈结构。
方法
在设计上使用两个栈,一个栈用来保存当前栈中的元素,其功能和一个正常的栈
没有区别,这个栈记为stackData; 另一个栈用于保存每一步的最小值,这个栈记为stackMin。
举例
stackData存放3 4 5 1 2 1
则stackMin存放3 1 1
代码实现
import java.util.Scanner;
public class MyStack {
private Stack<Integer> stackData;
private Stack<Integer> stackMin;
public static void main(String[] args) {
Scanner cs = new Scanner(System.in);
MyStack stack = new MyStack();
System.out.println("输入数据个数:");
int n = cs.nextInt();
for(int i=0;i<n;i++) {
int temp = cs.nextInt();
stack.push(temp);
}
stack.pop(); //只为测试pop函数
System.out.println(stack.getMin());
}
public MyStack()
{
stackData = new Stack<Integer>();
stackMin = new Stack<Integer>();
}
void push(int m) {
stackData.push(m);
if(stackMin.empty()) stackMin.push(m);
else {
if(stackMin.peek()>=m)
stackMin.push(m);
//stackMin栈顶元素大于等于m时加入
}
}
//此后当有元素出栈时,stackData正常出栈,如若出栈元素等于stackMin的栈顶,
//则stackMin栈顶一同出栈,保证实时更新stackMin
void pop() {
if(stackData.empty())
throw new RuntimeException("Your empty is empty!");
int a = stackData.pop();
if(a == stackMin.peek()) stackMin.pop();
}
int getMin() {
if(stackData.empty())
throw new RuntimeException("Your empty is empty!");
int a = stackMin.peek();
return a;
}
}
说明:只为记录自己每日学习,加油 ~