算法-包含min函数的栈

题目描述

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

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();
	}
}

思路：

搞一个辅助栈，来存放当前数据栈的最小值

一、push操作

不管什么情况先入数据栈，然后看最小值栈是不是为空，为空就入栈，或者当前值入栈的值小于最小栈顶部的值，也入最小栈，如果当前值大于最小栈栈顶的值，当前值就不入最小值栈，而是把最小值栈顶元素入栈（这里是为了保证两个栈的元素个数是一致的，这样获取最小值是，才是当前数据栈的最小值）

二、pop操作

数据栈和最小值栈直接出栈

三、顶部元素

就是数据栈的顶部元素

四、min操作

最小值栈的栈顶一定当前数据栈中所有元素中最小的