【剑指Offer】用两个栈实现队列&&用两个队列实现栈

用两个栈实现队列

题目描述

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

思路：

      栈是先进后出的数据结构，而队列是先进先出。为了实现队列，那么我们必须将数据先压入其中一个栈（栈A）中，数据压入完毕之后，将数据都出栈，并压入另一个栈（栈B）中，压入完毕之后栈B中元素出栈，就达到了先进先出的效果。

      那么我们在push时所需要做的工作就是把数据压入栈A

      在pop时，首先判断两个栈是否为空，如果都为空则抛出异常。如果栈B不为空，那么栈B元素出栈，如果栈B为空，将栈A元素都出栈并压入栈B中，压入完毕之后，将栈B元素出栈。

代码

import java.util.Stack;



public class Solution {

    Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 = new Stack<Integer>();

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

    public int pop() {
        //如果栈都为空，抛出异常
        if(stack1.empty() && stack2.empty()){
            throw new RuntimeException("Queue is empty");
        }

        //如果栈B为空，那么将栈A的元素都出栈并压入栈B
        if(stack2.empty()){
            while(!stack1.empty()){
                int t = stack1.pop();
                stack2.push(t);
            }
        }

        //栈B元素一一出栈
       return stack2.pop();
    }
}

拓展：两个队列实现栈

用两个队列实现栈更困难一些，因为队列是先进先出，你会发现如果把数据一起操作，无论怎样都是先进先出的顺序。例如：将1、2、3分别入队（队A），再出队，然后压入另一个队列（队B），出队顺序还是1、2、3。所以我们必须要把数据分开操作。对于刚刚例子中，我们期望的出队数字为3，所以我们将1、2从队A出队，并放入队B中，然后队A还剩下3，我们将3出队并存入变量c中，再将1、2出队B，放回队A中。返回值为c。简单来说每次pop时，都将队A元素出队，直到只剩下一个元素为止。

import java.util.PriorityQueue;
import java.util.Queue;

class Solution {
    Queue<Integer> q1 = new PriorityQueue<Integer>();
    Queue<Integer> q2 = new PriorityQueue<Integer>();
    
    public void push(int node) {
        q1.offer(node);
    }
    
    public int pop() {
        //如果队列都为空，抛出异常
        if(q1.isEmpty() && q2.isEmpty()){
            throw new RuntimeException("Stack is empty");
        }
        int t=0;
        //如果队A中只有一个元素，那么出队
        if(q1.size()==1)t = q1.poll();
        //如果队A中不止一个元素，那么将元素出队并压入队B，直到队A只剩下一个元素
        else if(q1.size() > 1){
            while(q1.size() > 1){
                q2.offer(q1.poll());
            }
            //将队A中剩下的元素出队并存入t中
            t = q1.poll();
            //将其他元素再放回队A中，以便下次使用。将队B中元素出队并压入队A
            while(!q2.isEmpty()){
                q1.offer(q2.poll());
            }
        }
       return t;
    }
}