剑指offer之编程（十六）

题目描述：

定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的min函数，在该栈中，调用min，pop，push的时间复杂度都是O(1)。

思路：

首先要注意题目的要求，是定义栈的数据结构，写一个函数min得到栈对的最小元素，也就是说，不管栈是怎么变的，这个函数得到的都是当前栈的最小元素，并不是弹出栈中最小元素等操作（由于讨论时有同学提出疑问，问题就是理解错了题目，既然是剑指offer，那么理解错了题目就相当于面试失败了一半）。对于这道题，首先想到的是使用一个临时的变量在入栈时存放栈的最小元素，这种做法的错误之处在于不能实时得到栈的最小元素，即时使用遍历栈的方法来更新这个值也会使时间复杂度不符合要求。因此，应该使用一个辅助的栈来进行最小值的存储，其实，在栈的操作中，很多地方都需要使用辅助的栈来进行相关操作。

使用辅助栈的思路如下： 当第一个数入栈时，这个数是当前栈中最小的元素，同时压入辅助栈，当第二个数入栈时，判断这个数与辅助栈中的最小的元素（即栈的最上面的元素）的大小，当大于这个最小元素时，就将这个最小元素再压入一次辅助栈，当小于这个最小元素时，就压入辅助栈。这里辅助栈和储存数据的栈的入栈和出栈是同时进行的，只是入栈的元素有所区别而已。

代码如下：

import java.util.*;

public class Main23 
{
	
	
	public static void main(String[] args)   //随便写的一个测试程序
	{
		MinStack ms = new MinStack();
		Scanner s = new Scanner(System.in);
		int n = 4;  //预计输入栈的个数
		
		for(int i=0;i<n;i++)  //依次入栈
		{
			ms.push(s.nextInt());
		}
		
		System.out.println(ms.min());  //打印最小元素
		ms.pop();  //弹出一个元素
		System.out.println(ms.min());  //打印最小元素
		ms.pop();  //弹出一个元素
		System.out.println(ms.min());  //打印最小元素
		ms.pop();  //弹出一个元素
		System.out.println(ms.min());  //打印最小元素
		
	}

	
	static class MinStack
	{
		Stack<Integer> data = new Stack<Integer>();  //储存数据的栈
		Stack<Integer> temp = new Stack<Integer>();  //储存最小元素的栈
		
		public void push(int num)  //压入一个元素
		{
			data.push(num);
			
			if(temp.size()==0 || num<temp.peek())  //当是第一个元素或小于最小元素时，直接压入辅助栈
				temp.push(num);
			else
				temp.push(temp.peek());  //再次压入最小元素
		}
		
		public void pop()  //弹出一个元素
		{
			if(data.size()>0 && temp.size()>0)  //存储数据的栈和辅助栈动作要一致，一起弹出
			{
				data.pop();
				temp.pop();
			}
		}
		
		public int min()  //返回当前栈中最小元素，即直接取辅助栈的栈顶元素
		{
			return temp.peek();  //Java原本所带的栈会有判断栈是否为空并抛出Exception，这里暂且不处理 
		}
		
		public int top()  //返回栈顶元素但不弹出
		{
			return data.peek();  //Java原本所带的栈会有判断栈是否为空并抛出Exception，这里暂且不处理
		}
	}
	
}

本人经验，仅供参考！