返回栈中的最小值

最新推荐文章于 2021-06-27 22:41:58 发布

原创最新推荐文章于 2021-06-27 22:41:58 发布 · 389 阅读

0 ·

CC 4.0 BY-SA版权

算法专栏收录该内容

46 篇文章

订阅专栏

本文介绍了一种使用两个栈来实现数据结构中快速获取最小元素的方法。通过维护一个数据栈和一个最小值栈，可以在O(1)的时间复杂度内获取到当前栈中的最小值，同时支持push和pop操作。

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

用另一个栈存放栈的最小值。

import java.util.Stack;


public class Minstack {

	Stack<Integer> minStack;
	Stack<Integer> dataStack;
	public Minstack() {
		this.minStack=new Stack<>();
		this.dataStack=new Stack<>();
	}
	public void push(int num) {
		dataStack.push(num);
		if(minStack.isEmpty()||minStack.peek()>=num)
			minStack.push(num);
		else
			minStack.push(minStack.peek());
	}
	public Integer pop() {
		if(dataStack.isEmpty())
		{
			throw new RuntimeException("dataStack is Empty!");
		}
		if(dataStack.peek()==minStack.peek())
		{
			dataStack.pop();
			return minStack.pop();
		}
		else
		{
			return dataStack.pop();
		}
		
	}
	public int getMin() {
		if(!minStack.isEmpty())
			return minStack.peek();
		throw new RuntimeException("minStack is Empty!");
	}
	public static void main(String[] args) {
		Minstack mstack=new Minstack();
		mstack.push(3);
		mstack.push(5);
		mstack.push(2);
		mstack.push(1);
		
		System.out.println(mstack.getMin());
		mstack.pop();
		System.out.println(mstack.getMin());
		
	}

}