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


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

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

思路

采用两个栈:

  stackData作为数据栈保存数据;

  stackMin作为辅助栈保存stackData数据栈中每一步的最小元素。

压入数据push操作:

 1.判断stackMin辅助栈是否为空,
          若为空,将newNum同步压入stackMin辅助栈,执行下一步;

          若不为空,直接执行下一步;

2.判断newNUm是否小于此时的stackMin.peek(),

        若小于,将newNum压入stackMin中。否则,执行下一步3。

3.判断newNUm是否大于等于此时的stackMin.peek()(代码中用getmin()函数表示),

        若大于等于,将此时的stackMin赋值给新变量newMin,并将newMin压入进stackMin。

4.stackData同时一直压入newNum,结束操作。


弹出数据pop操作:

 1.判断stackMin是否为空,

        若为空,提示报错。执行下一步。

2.直接弹出stackMin栈顶元素。

3.返回stackData栈顶元素。

代码

public static class MyStack1 {
		private Stack<Integer> stackData;
		private Stack<Integer> stackMin;

		public MyStack1() {
			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 { //newNum >= this.getmin())压入this.stackMin.peek()
				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();
		}
	}

	public static void main(String[] args) {
		MyStack1 stack1 = new MyStack1();
		stack1.push(3);
		System.out.println(stack1.getmin());
		stack1.push(4);
		System.out.println(stack1.getmin());
		stack1.push(1);
		System.out.println(stack1.getmin());
		System.out.println(stack1.pop());
		System.out.println(stack1.getmin());
	}

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值