牛客刷题之包含min函数的栈

题目描述

定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的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;
    }

}