题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
解题思路:
由于java中栈的底层也是通过数组实现的,所以本题用数组来实现栈的基本操作。
首先,定义一个变量来保存栈中的最小元素;
(1)栈的push操作:通过往数组中添加元素来实现栈的push操作。注意判断是否是第一次压栈,如果是的话将第一次压栈的元素赋值给最小值变量num,然后添加到数组中,以后往数组中添加的元素都要和这个最小变量num进行比较,保证num中存在的永远是最小的值。
(2)栈的pop操作:分三种情况:>如果数组此时为空,什么都不做;
>如果出栈的恰好是目前最小的数,则需要重新在数组中寻找次小的数,找到之后,更新num 的值;
>否则,直接从数组中删除即可。
(3)栈的top操作:直接返回数组中的最后一个元素即可。
(4)栈的min:直接返回num即可得到最小元素。
import java.util.*;
public class Solution {
ArrayList<Integer> arr = new ArrayList<Integer>();
int num;
public void push(int node) {
if(arr .size()==0){
num = node;
}
arr.add(node);
if(node < num){
num = node;
}
}
public void pop() {
if(arr .size()==0){
return;
}
if(arr.get(arr.size()-1)==num)
{
int temp_min=arr.get(0);
for(int i=0;i<arr.size()-1;i++){
if(arr.get(i)<temp_min){
temp_min = arr.get(i);
}
}
num=temp_min;
}
arr.remove(arr.size()-1);
}
public int top() {
if(arr .size()==0){
return 0;
}
return arr.get(arr.size()-1);
}
public int min() {
return num;
}
}