剑指offer_4_两个栈实现队列

两个栈实现队列


/**
 * 用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型
 */

思路:

有两个栈,一个专门负责进入,另一个专门负责输出,当遇到pop时,如果负责输出,不为空,直接pop,如果负责输出的为空,则将负责进入的出栈,并压入负责输出的栈,然后直接pop,
static class queue{
       Stack<Integer> stack1 = new Stack<Integer>();
       Stack<Integer> stack2 = new Stack<Integer>();


       public void push(int node) {
           stack1.add(node);
       }

       public int pop() {
           if(!stack2.isEmpty()){
               return stack2.pop();
           }
           else {
               while(!stack1.isEmpty()){
                   stack2.add(stack1.pop());
               }
               return stack2.pop();
           }
       }
   }
发布了63 篇原创文章 · 获赞 1 · 访问量 4128

猜你喜欢

转载自blog.csdn.net/chenhanhao0000/article/details/104053403