实现O(1)获取最大最小值的栈和队列----java
一.如何实现包含获取最小值函数的栈
问题:定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的getMin函数。在该栈中,调用getMin、push及pop的时间复杂度都是O(1).
最小值思路:用一个辅助栈stack2记住每次入栈stack1的当前最小值:在stack1入栈时,往stack2中加入当前最小值;stack1元素出栈时,stack2也出栈一个元素。最小值从stack2中获取及栈顶元素。O(1)
最大值思路:同上O(1)
中间值思路:对栈排序,取中间值,但是时间不是O(1)
package com.sheepmu; import java.util.Arrays; import java.util.Stack; public class SpecialStack<E extends Comparable<E>> { Stack<E> stack1=new Stack<E>(); Stack<E> stackMin=new Stack<E>();//存放求最小值的栈 Stack<E> stackMax=new Stack<E>();//存放求最大值的栈 public void push(E e) { stack1.push(e); if(stackMin.isEmpty()||e.compareTo(stackMin.peek())<0) stackMin.push(e);//若最小栈为空push进stack时就同时把它push进stackMin; else if(e.compareTo(stackMin.peek())>0) stackMin.push(stackMin.peek()); if(stackMax.isEmpty()||e.compareTo(stackMin.peek())>0) stackMax.push(e); else if(e.compareTo(stackMax.peek())<0) stackMin.push(stackMax.peek()); } public E pop()//一定要记着,非空才能pop() { if(!stack1.isEmpty()&&!stackMin.isEmpty()&&!stackMax.isEmpty()) { E e=stack1.pop(); stackMin.pop(); stackMax.pop(); return e; } else { System.out.println("栈已经为空了"); return null; } } public E getMin() { return stackMin.peek(); } public E getMax() { return stackMax.peek(); } public E getMed() { E[] ss=(E[])stack1.toArray();//stack1.toArray()返回的是Object[], 栈----->数组 Arrays.sort(ss); return ss[ss.length/2]; } }
实现O(1)获取最大最小值的栈----java,布布扣,bubuko.com
时间: 2024-10-05 04:41:15