剑指offer-面试题21-包含min函数的栈

package case21_MyStackWithMin;

import java.util.Stack;

/**
 * 题目:定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的min函数。在该栈中,调用min,push,pop的时间复杂度都是O(1)。
 * 
 * @author WangSai
 *
 */
public class MyStackWithMin {
	/**
	 * 1,思路:通过两个栈实现,数据栈stackData实现存放数据,辅助栈stackMin实现存放最小的数据。
	 * 2,压栈:只要有非空元素请求入栈则压入stackData。同时判断该元素与辅助栈stackMin的栈顶元素的大小,若小于等于stackMin栈顶元素,则入栈stackMin;
	 * 否则,不压栈。stackMin的元素个数 不大于 stackData。
	 * 3,弹栈,stackData不为空则立马弹栈;判断stackMin的栈顶元素是否与stackData刚才的出栈元素相同,如果相同则stackMin栈顶元素也出栈。
	 *
	 * @param <T>,泛型定义数据类型
	 */
	public static class MyStack<T extends Comparable<T>> {
		private Stack<T> stackData;
		private Stack<T> stackMin;

		// 构造函数
		public MyStack() {
			this.stackData = new Stack<>();
			this.stackMin = new Stack<>();
		}

		// 实现压栈
		public void push(T t) {
			// 异常值检测,t不能为null
			if (t == null)
				throw new IllegalArgumentException("IllegalArgument");
			// 如果两个栈都为null,则将t分别压入两个栈。
			if (stackData.empty()) {
				stackData.push(t);
				stackMin.push(t);
			}
			// 如果新元素 小于等于 stackMin的栈顶元素,则压入栈stackMin中
			else if (t.compareTo(stackMin.peek()) <= 0) {
				stackMin.push(t);
				stackData.push(t);
			}
			// 只要有新元素,就需要压入stackData中
			else {
				stackData.push(t);
			}
		}

		// 实现弹栈
		public T pop() {
			if (stackData.empty())
				throw new RuntimeException("Stack is already empty");
			// 只要有弹栈需求,就把stackData栈顶元素弹出去
			T t = stackData.pop();
			// 如果栈stackData不为null,则stackMin中肯定有元素。
			// 若pop出的stackData中的元素,恰好也是是stackMin的栈顶元素,则删除stackMin的栈顶元素
			if (t != null && t == stackMin.peek()) {
				stackMin.pop();
			}
			return t;
		}

		// 获取最小值
		public T min() {
			if (stackData.empty())
				throw new IllegalArgumentException("Stack is already empty");
			return stackMin.peek();
		}
	}

	// 测试MyStack
	public static void main(String[] args) {
		MyStack<Integer> stack1 = new MyStack<>();
		// 压入3,获取min
		stack1.push(3);
		System.out.println("压入3,获取min: " + stack1.min());
		// 压入4,获取min
		stack1.push(4);
		System.out.println("压入4,获取min: " + stack1.min());
		// 压入5,获取min
		stack1.push(5);
		System.out.println("压入5,获取min: " + stack1.min());
		// 压入0,获取min
		stack1.push(0);
		System.out.println("压入0,获取min: " + stack1.min());
		// pop一次,弹出0之后,获取min应该是3
		System.out.println("pop第1次:       	    " + stack1.pop());
		System.out.println("pop第1次之后,获取min: " + stack1.min());

		System.out.println("pop第2次:       	    " + stack1.pop());
		System.out.println("pop第3次:       	    " + stack1.pop());
		System.out.println("pop第3次之后,获取min: " + stack1.min());
		System.out.println("pop第4次:       	    " + stack1.pop());
		System.out.println("pop第5次: " + stack1.pop());
		System.out.println("pop第5次之后,获取min: " + stack1.min());
	}
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值