用两个队列实现栈

思路:

  1. 先往 queue1 顺次插入1,2,3,4,5,此时按照栈的规则应先出来 5,所以先将1,2,3,4 出队列 queue1,并入队列 queue2,5 出队列,queue2 当前状态是:1,2,3,4,queue1当前状态为空;
  2. 将1,2,3入队列 queue1,4 出队列,此时,queue2 为空,queue1 有 1,2,3...
  3. 重复以上步骤

代码实现:

/**
 * 题目类型:队列
 *
 * 题目要求:用两个队列实现栈
 *
 * 思路:1. 先往 queue1 顺次插入1,2,3,4,5,此时按照栈的规则应先出来 5,所以先将1,2,3,4 出队列 queue1,并入队列 queue2,5 出队列
 *  queue2 当前状态是:1,2,3,4,queue1当前状态为空;
 *  2. 将1,2,3入队列 queue1,4 出队列,此时,queue2 为空,queue1 有 1,2,3...
 *  3. 重复以上步骤
 */
public class QueueToStack {

    private Queue<Integer> queue1 = new ArrayDeque<Integer>();

    private Queue<Integer> queue2 = new ArrayDeque<Integer>();

    /**
     *  向栈中压入数据
     * @param data
     */
    private void push(Integer data) {
        if (queue1.isEmpty() && queue2.isEmpty()) {
            queue1.add(data);
        }

        //如果queue1为空,queue2有数据,直接放入queue2
        if (queue1.isEmpty()) {
            queue2.add(data);
        }

        if (queue2.isEmpty()) {
            queue1.add(data);
        }
    }

    /**
     * 从栈中弹出数据
     * @return
     */
    private Integer pop() throws Exception {
        if (queue1.isEmpty() && queue2.isEmpty()) {
            throw new Exception("stack is empty");
        }

        //如果queue1中没有元素,queue2中有元素,将queue2中的元素依次放入queue1中,弹出最后一个元素
        if(queue1.isEmpty()){
            while(queue2.size() > 1){
                queue1.add(queue2.poll());
            }
            return queue2.poll();
        }

        //如果queue2中没有元素,queue1中有元素,将其queue1中的元素依次放入queue2中,弹出最后一个元素
        if(queue2.isEmpty()){
            while(queue1.size() > 1){
                // poll() Retrieves and removes the head of this queue 若队列为空,则返回 null
                // peek() Retrieves, but does not remove, the head of this queue
                queue2.add(queue1.poll()); 
            }
            return queue1.poll();
        }

        return null;
    }

    @Test
    public void testQueue() throws Exception {
        QueueToStack queueToStack = new QueueToStack();
        queueToStack.push(1);
        queueToStack.push(0);
        queueToStack.push(6);
        queueToStack.push(9);
        System.out.println(queueToStack.pop());
        System.out.println(queueToStack.pop());

        queueToStack.push(2);
        System.out.println(queueToStack.pop());
        System.out.println(queueToStack.pop());
        System.out.println(queueToStack.pop());
    }
}
复制代码
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值