剑指offer第二版面试题9:用两个队列实现栈(JAVA版)

题目:用两个队列实现栈。spa

分析:经过一系列的栈的压入和弹出操做来分析用队列模拟一个栈的过程,如图所示,咱们先往栈内压入一个元素a。因为两个队列如今都是空,咱们能够选择把a插入两个队列中的任一个。咱们不妨把a插入queue1。接下来继续往栈内压入b,c两个元素。咱们把它们都插入queue1。这个时候 queue1包含3个元素a,b,c其中a位于队列的头部,c位于队列的尾部。code

如今咱们考虑从栈内弹出一个元素。根据栈的后进先出的原则,最后被压入栈的c应该最早被弹出。因为c位于queue1的尾部,而咱们每次只能从队列的头部删除元素,所以咱们能够从queue中依次删除a/b并插入到queue2中,再从queue1中删除c。这就至关于从栈中弹出元素c了。咱们能够用一样的方法从栈内弹出元素b。blog

接下来咱们考虑从栈内压入一个元素d.此时queue1已经有了一个元素,咱们就把d插入到queue1的尾部。若是咱们再从栈内弹出一个元素,此时被弹出的应该是最后被压入的d.因为d位于queue1的尾部,咱们只能先从头部删除 queue1的元素并插入到queue2,直到queue1中遇到d再直接把它删除。如图所示:队列

代码以下:rem

public class MyStack {
    private Queue<Integer> queue1;
    private Queue<Integer> queue2;
    public MyStack(){
        queue1=new LinkedList<>();
        queue2=new LinkedList<>();
    }
    
    //添加,哪一个队列不是空则添加到那个队列
    public void push(Integer num){
        if(queue1.size()==0 && queue2.size()==0){
            queue1.add(num);
        }else if(queue1.size()==0){
            queue2.add(num);
        }else{
            queue1.add(num);
        }
    }
    
    //删除
    public Integer pop(){
        if(queue1.size() == 0 && queue2.size() == 0){
            return null;
        }
        int value=0;
        if(queue1.size()>0){
            while(queue1.size()>0){
                queue2.add(queue1.remove());
            }
            value=queue1.remove();
        }
        if(queue2.size()>0){
            while(queue2.size()>0){
                queue1.add(queue2.remove());
            }
            value=queue2.remove();
        }
        return value;
    }
}
相关文章
相关标签/搜索