剑指offer-包含min函数的栈

剑指offer-包含min函数的栈

题目描述

定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数（时间复杂度应为O（1））。

解题思路

这道题乍一看题目描述可能会有点懵，反正我是没看懂是想要干什么，但是描述一般会把问题描述的很清楚，这里就是找到一个时间复杂度为o(1)的min函数，一般找最小值，首先想的就是for循环，但是如果是o(1)的话，那就是直接就能找到，不能做遍历，所以要想办法直接拿到这个最小值。再加上题目中要求栈的数据结构，栈的栈顶我们可以直接拿到吧，所以想到一个辅助栈，每次存入比较小的数，然后栈是先进后出，所以此时栈顶元素就是最小值，然后既然要模拟栈，那就要有出入栈操作，所以，总体思路是：

应用一个辅助栈，压的时候，如果A栈的压入比B栈压入小于等于，AB栈同时压入;如果A栈的压入比B栈压入大，A栈压，B栈压入自己的栈顶元素，即上一次比较小的那个元素，这样保证AB栈长度一致，，，，这样出栈时AB栈同时出。

代码如下：

import java.util.Stack;

public class Solution {

	Stack<Integer> dataStack = new Stack<Integer>();
    Stack<Integer> minStack = new Stack<Integer>();

    public void push(int node) {
        dataStack.push(node);
        if(minStack.isEmpty() || node < minStack.peek()){
        	minStack.push(node);
        }else{
        	minStack.push(minStack.peek());
        }
    }
    
    public void pop() {
        dataStack.pop();
        minStack.pop();
    }
    
    public int top() {
        return dataStack.peek();
    }
    
    public int min() {
        return minStack.peek();
    }
}