劍指offer 第二十題 包含min函數的棧 java

題目描述

定義棧的數據結構，請在該類型中實現一個能夠得到棧中所含最小元素的min函數（時間複雜度應爲O（1））。

 

思路：

       除了正常的Stack之外，需要一個額外的排序數組（也可以是Stack）

import java.lang.reflect.Array;
import java.util.Stack;

public class Solution {
    Stack<Integer> stack = new Stack<Integer>();
    int []array = new int[1000];
    int length =0;
    public void push(int node) {

        array[length] =node;
        length++;
        stack.push(node);
        if(length >=2){
            for(int i=0;i<length-1;i++){
                for(int j=i+1;j<=length-1;j++){
                    if(array[i]<array[j]){
                        int tem =array[i];
                        array[i] = array[j];
                        array[j] =tem;
                    }
                }
            }
        }
    }

    public void pop() {
        int num =stack.peek();
        int step=0;
        for(int i=0;i<=length-1;i++){
            if(array[i] == num){
                step =i;
                break;
            }
        }
        for(int i =step;i<length-1;i++)
            array[i] =array[i+1];
        array[length-1] =0;
        length--;
        stack.pop();
    }

    public int top() {
        return stack.peek();
    }

    public int min() {
        return array[length-1];
    }
}