【剑指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;
    }
}

 

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值