剑指offer-21-包含min函数的栈-java

题目及测试

package sword021;
/* 题目描述：定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的min函数。在该栈中，调用min、push以及pop的时间复杂度为O(1)

*/
public class main {
	
	public static void main(String[] args) {
		test();
	}
		 
	private static void test() {
		Solution solution = new Solution();
		int rtn;
		long begin = System.currentTimeMillis();
		solution.push(1);
		solution.push(2);
		solution.push(0);
		
		long end = System.currentTimeMillis();		
		System.out.print(solution.min()+" ");
		System.out.println();
		System.out.print(solution.top()+" ");
		System.out.println();
		solution.pop();
		System.out.print(solution.min()+" ");
		System.out.println();
		System.out.print(solution.top()+" ");
		System.out.println();
		System.out.println("耗时：" + (end - begin) + "ms");
		System.out.println("-------------------");
	}

}

解法1（成功）

首先这个栈要具有普通栈所具有的push()和pop()方法，那么内部一定包含一个Stack。至于还要能实现min函数，而且还是在O(1)时间复杂度内，我们不得不考虑用额外的空间。

如果直接使用一个int变量存储当前的最小值，我们的确可以获得最小值，但是当栈pop()了以后，我们无法获得次小值。我们需要一个数据结构来动态保存每个时刻的最小值，每当push()和pop()的时候，同样更新这个数据结构，而且时间复杂度也是O(1)，那么看来需要额外O(n)的空间了。可以选择栈或者一个链表（实质上当作栈使用）。我们使用一个辅助栈来实现，该栈和主栈的大小相同，栈顶存放的是当前栈内的最小值。

package sword021;

import java.util.Stack;


public class Solution {

    //数据栈
    private Stack<Integer> stackData;
    //辅助栈
    private Stack<Integer> stackMin;

    public Solution(){
        stackData=new Stack<Integer>();
        stackMin=new Stack<Integer>();
    }

    //入栈操作
    public void push(int node) {
    	stackData.push(node);
    	if(stackMin.isEmpty() || node < min()) {
    		stackMin.push(node);
    	}else {
    		stackMin.push(min());
    	}
    }
    //出栈操作
    public void pop() {
    	stackData.pop();
    	stackMin.pop();
    }
    //返回栈顶的元素
    public int top() {
        return stackData.peek();
    }
    //得到栈的最小元素
    public int min() {
        return stackMin.peek();
    }
}