返回栈中最小元素问题

问题是这样的:

实现一个特殊的栈,在实现栈的基本功能的基础上,再实现返回栈中最小元素的操作。要求:1.pop、push、getMin操作的时间复杂度都是O(1)。2.设计的栈类型可以使用现成的栈结构。

这个问题可以用两个栈来实现。一个栈用来存放要push和pop的元素,另一个栈用来返回最小值。具体操作是当push一个数时,先放到栈1中,栈2中如果没有元素就直接放入,有元素就和栈顶元素比较,小于或等于栈顶元素就直接放入,大于时就放入一个等于栈顶元素的值,那么栈顶元素一定是最小元素,而且pop时,左右栈一起pop,之后栈2的栈顶仍然是栈内的最小元素。

代码如下:

public class GetMinStack {
	public static void main(String[] args) {
		MyStack stack = new MyStack();
		stack.push(3);
		System.out.println(stack.getmin());
		stack.push(4);
		System.out.println(stack.getmin());
		stack.push(1);
		System.out.println(stack.getmin());
		System.out.println(stack.pop());
		System.out.println(stack.getmin());
	}
	public static class MyStack {
		private Stack<Integer> stackData;
		private Stack<Integer> stackMin;
		public MyStack() {
			this.stackData = new Stack<Integer>();
			this.stackMin = new Stack<Integer>();
		}
		public void push(int newNum) {
			if(this.stackMin.isEmpty()) {
				this.stackMin.push(newNum);
			} else if(newNum < this.getmin()) {
				this.stackMin.push(newNum);
			} else {
				int newMin = this.stackMin.peek();
				this.stackMin.push(newMin);
			}
			this.stackData.push(newNum);
		}
		public int pop() {
			if(this.stackData.isEmpty()) {
				throw new RuntimeException("Your stack is empty.");
			}
			this.stackMin.pop();
			return this.stackData.pop();
		}
		//获取最小值
		public int getmin() {
			if (this.stackMin.isEmpty()) {
				throw new RuntimeException("Your stack is empty.");
			}
			return this.stackMin.peek();
		}
	}
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值