（二十）包含min函数的栈

原创于 2019-06-02 11:06:10 发布 · 85 阅读

0 ·

CC 4.0 BY-SA版权

剑指offer（java）专栏收录该内容

26 篇文章

订阅专栏

本文介绍了一种特殊栈的数据结构实现，该栈包含一个能在O(1)时间复杂度内返回栈中最小元素的min函数。通过维护两个栈，一个用于常规元素的压入和弹出操作，另一个用于跟踪最小值，确保了min函数的高效性。

题目描述

定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数（时间复杂度应为O（1））。

import java.util.Stack;

public class Solution {
    Stack<Integer> stack = new Stack<>();
	Stack<Integer> temp = new Stack<>();
	int min  = Integer.MAX_VALUE;
	public void push(int node) {
		if(node < min){
			min = node;
			stack.push(node);
			temp.push(node);
		}else{
			stack.push(node);
			temp.push(min);
		}
	}

	public void pop() {
		stack.pop();
		temp.pop();
	}

	public int top() {
		int t = stack.pop();
		stack.push(t);
		return t;
	}

	public int min() {
		int t = temp.pop();
		temp.push(t);
		return t;
	}
}