知识点/数据结构:栈和队列
题目描述:
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
中间发现的问题:
2018.12.5自己写了测试案例,发现了有错误!!!!在下图的最后一行,原来写的是num=node;竟然也通过了,库中的测试案例不够。例如压入如下数字:5,2,4,3。这时候调用min()输出是3;有错,需要改为下图最后一行num=min();
借助一个辅助栈;代码如下:
package 包含min函数的栈;
import java.util.Stack;
public class Solution{
Stack<Integer> data=new Stack<Integer>();
Stack<Integer> min = new Stack<Integer>();
Integer num=null;//这是一个包装类对象;
public void push(int node){//这里有int到它的包装类Integer的自动装箱
if(num==null){//入栈第一次入栈,两个全进,第二次开始判断小的才入栈
data.push(node);
min.push(node);
num=node;
}else{
if(node<=num){
min.push(node);
}
data.push(node);
num=min();//这个很多次丢掉了
}
}
public void pop(){
int num1= data.pop();
int num2= min.pop();
if(num1!=num2){
min.push(num2);
}
}
public int top(){
int num= data.pop();
data.push(num);
return num;
}
public int min(){
int num = min.pop();
min.push(num);
return num;
}
}
测试代码如下:
package 包含min函数的栈;
public class Test {
public static void main(String[] args) {
Solution s = new Solution();
s.push(5);
s.push(2);
s.push(4);
s.push(3);
System.out.println(s.min());
}
}
运行结果如下: