题目:
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
书中的思路:
按照这个思路我们很容易写出以下代码:
import java.util.Stack; public class Solution { Stack stack = new Stack(); Stack assistStack = new Stack(); public void push(int node) { stack.push(node); if(assistStack.isEmpty()){ assistStack.push(node); }else if((int)assistStack.peek()>node){ assistStack.push(node); }else{ assistStack.push((int)assistStack.peek()); } } public void pop() { stack.pop(); assistStack.pop(); } public int top() { return (int)stack.peek(); } public int min() { return (int)assistStack.peek(); } }
设置一个辅助栈,存放当前最小的值。
原文地址:https://www.cnblogs.com/figsprite/p/10472325.html
时间: 2024-10-10 17:09:43