设计包含 min 函数的栈（栈） java语言实现

电玩女神 2022-08-12 00:16 202阅读 0赞

//定义栈的数据结构，要求添加一个 min 函///数，能够得到栈的最小元素。
    //要求函数 min、push 以及 pop 的时间复///杂度都是 O(1)。
    //在栈里添加一个成员变量存放最小元素（或//者最小元素的位置，考虑到元素可以是其他//复杂的数据类型，因此存在最小元素的位置//更能减少存储空间）。每次push一个新元///素进栈的时候，如果该元素比当前的最小元//素还要小，则更新最小元素。但这个思路存//在一个重要的问题，如果当前最小元素被p///op出去，如何才能得到下一个最小元素？
    
    //因此仅仅添加一个成员变量存放最小元素是//不够的，我们需要一个辅助栈，每次push///一个新元素的时候，同时将最小元素push///到辅助本中；每次pop一个元素出栈的时候//，同时pop辅助栈。
    
    
    import java.util.ArrayList;
    
    
    public class MinStack 
    {
    	ArrayList arrayStack=new ArrayList();
    	ArrayList min=new ArrayList();
    	public static void main(String args[])
    	{
    		MinStack stack=new MinStack();
    		stack.push(4);
    		stack.push(6);
    		stack.push(2);
    		stack.push(5);
    		stack.pop();
    		stack.pop();
    		stack.push(1);
    		System.out.print(stack.min());
    	}
    	private void push(Object obj)
    	{
    		//存放最小元素的位置
    		if(arrayStack.size()==0)
    		{
    			min.add(0);
    		}
    		
    		else if((Integer)obj<=(Integer)arrayStack.get((Integer)min.get(min.size()-1)));
    		{
    			min.add(arrayStack.size());
    		}
    		arrayStack.add(obj);	
    	}
    	private void pop()
    	{
    		if(arrayStack.size()-1==(Integer)min.get(min.size()-1))
    		{
    			arrayStack.remove(arrayStack.size()-1);
    			min.remove(min.size()-1);
    		}
    		else
    		{
    			arrayStack.remove(arrayStack.size()-1);
    		}
    	}
    	private Object min()
    	{
    		return (Integer)arrayStack.get((Integer)min.get(min.size()-1));
    		
    	}
    }