栈和队列----设计一个有getMin功能的栈

最新推荐文章于 2024-10-31 20:21:55 发布

原创最新推荐文章于 2024-10-31 20:21:55 发布 · 112 阅读

0 ·

CC 4.0 BY-SA版权

程序员代码面试指南专栏收录该内容

26 篇文章

订阅专栏

本文介绍了一种使用两个栈来实现数据结构，其中一个栈用于存储数据，另一个栈用于跟踪并存储当前栈中的最小值。这种方法可以高效地在任何时候获取到栈中的最小元素，而不需要遍历整个栈。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

关键点：

准备两个栈。一个数据栈stackData，一个存储当前栈中最小元素的栈stackMin。
stackMin的栈顶元素永远是当前栈的最小元素

代码：

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 data) {
		 stackData.push(data);
		 if(stackMin.isEmpty()) {
			 stackMin.push(data);
		 }else {
			if(data <= stackMin.peek()) {
				stackMin.push(data);
			}
		 }
		 
	 }
	 
	 public int pop() {
		 if(stackData.isEmpty()) {
			 throw new RuntimeException("Your stack is empty.");
		 }
		 
		 int value = stackData.pop();
		 if(stackMin.peek() == value) {
			 stackMin.pop();
		 }
		 return value;
		 
	 }
	 
	 
	 
	 public int getMin() {
		 if(stackMin.isEmpty()) {
			 throw new RuntimeException("Your stack is empty.");
		 }
		 return stackMin.peek();
	 }
	 
	
}
class MyStack2{
	private Stack<Integer> stackData;
	private Stack<Integer> stackMin;
	
	public MyStack2() {
		this.stackData = new Stack<Integer>();
		this.stackMin = new Stack<Integer>();
	}
	
	public void push(int data) {
		stackData.push(data);
		if(stackMin.isEmpty()) {
			stackMin.push(data);
		}else if(data <= stackMin.peek()){
			stackMin.push(data);
		}else {
			stackMin.push(stackMin.peek());
		}
	}
	
	public int pop() {
		if(stackData.isEmpty()) {
			throw new RuntimeException("Your stack is empty");
		}
		stackMin.pop();	
		return stackData.pop();
		
	}
	
	public int getMin() {
		if(stackMin.isEmpty()) {
			throw new RuntimeException("Your stack is empty");
		}
		return stackMin.peek();
	}
}