JZ20 包含min函数的栈

JZ20 包含min函数的栈

问题描述

定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数，并且调用 min函数、push函数 及 pop函数 的时间复杂度都是 O(1)
push(value):将value压入栈中
pop():弹出栈顶元素
top():获取栈顶元素
min():获取栈中最小元素

示例:
输入: [“PSH-1”,“PSH2”,“MIN”,“TOP”,“POP”,“PSH1”,“TOP”,“MIN”]
输出: -1,2,1,-1
解析:
"PSH-1"表示将-1压入栈中，栈中元素为-1
"PSH2"表示将2压入栈中，栈中元素为2，-1
“MIN”表示获取此时栈中最小元素==>返回-1
"TOP"表示获取栈顶元素==>返回2
"POP"表示弹出栈顶元素，弹出2，栈中元素为-1
"PSH-1"表示将1压入栈中，栈中元素为1，-1
"TOP"表示获取栈顶元素==>返回1
“MIN”表示获取此时栈中最小元素==>返回-1

问题分析

需要返回最小值，可以创建一个list来存储值并得到最大的值。
push：在压入栈的同时，在list中再存一份
pop：在弹出栈顶元素的同时删除list中的最后一个
top：获取栈顶元素不删除peek
min：将list转换为数组arr，对数组排序，取第一个数arr[0]

解法

import java.util.*;
public class Solution {
    List<Integer> list=new ArrayList<Integer>();
    Stack<Integer> sc=new Stack<Integer>();
    public void push(int node) {
        list.add(node);
        sc.push(node);
    }
    public void pop() {
        list.remove(list.size()-1);
        sc.pop();
    }
    public int top() {
        return sc.peek();
    }
    public int min() {
        Object[] arr = list.toArray();
        Arrays.sort(arr);
        return (Integer) arr[0];
    }
}