一个有getMin功能的栈

前言:这是学习《程序员代码面试指南》一书的第一篇文章,也是博客的第一篇文章,给自己加油,从现在开始坚持记录学习。



设计一个有getMin功能的栈

题目:实现一个特殊的栈,在实现栈的基本功能的基础上,再实现返回栈中最小值的操作。

要求:1.pop、push、getMin操作的时间复杂度都是O(1)。

            2.设计的栈类型可以使用现成的栈结构。


1.思路整理:

  采用两个栈,
     stackData作为数据栈保存数据;
     stackMin作为辅助栈保存stackData数据栈中每一步的最小元素。
 
   压入数据规则:
      1.将数据newNum压入stackData数据栈中,执行下一步;
      2.判断stackMin辅助栈是否为空,
          若为空,将newNum同步压入stackMin辅助栈,执行下一步;
          若不为空,直接执行下一步;
      3.若 newNum <= stackMin.peek(),将newNum同步压入stackMin,
         若 newNum > stackMin.peek(),结束操作
      
     弹出数据规则:
      1.判断stackData栈是否为空,
          若为空,结束操作;
          若不为空,弹出栈顶元素,执行下一步;
      2.判断stackMin辅助栈是否为空,
          若为空,结束操作
          若不为空,直接执行下一步;
      3.若 stackData栈顶元素 == getMin(),弹出stackMin栈顶元素
         若 stackData栈顶元素 > getMin(),结束操作
       (注:根据压入规则,stackData栈中元素 >= stackMin栈中元素,所以上面不必考虑小于的情况)
      
     查询最小值操作:
         根据以上规则, stackMin的栈顶元素始终是当前stackData栈的最小值
       

2.代码实现:


public class MyStack {
	private Stack<Integer> stackData;
	private Stack<Integer> stackMin;
	
	public MyStack() {
		this.stackData = new Stack<>();
		this.stackMin = new Stack<>();
	}
	
	public void push(int newNum) {
		if (this.stackMin.empty()) {
			this.stackMin.push(newNum);
		} else if(newNum<=this.getMin()){
			this.stackMin.push(newNum);
		}
		this.stackData.push(newNum);
	}
	
	public int pop() {
		if(this.stackData.empty()){
			throw new RuntimeException("操作的栈为空");
		}
		if(this.stackData.peek() == this.getMin()){
			this.stackMin.pop();
		}
		return this.stackData.pop();
	}
	
	public int getMin() {
		if(this.stackMin.empty()){
			throw new RuntimeException("操作的栈为空");
		}
		return this.stackMin.peek();
	}
	
}


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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值