java数据结构与算法之使用两个栈实现队列

Posted wen-pan

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了java数据结构与算法之使用两个栈实现队列相关的知识,希望对你有一定的参考价值。

题目要求

  • 要求使用两个栈来实现一个队列的功能
  • 实现addLast功能
  • 实现pollFirst功能
  • 实现peekFirst功能
  • 实现isEmpty功能

代码实现

/**
 * 使用两个栈来实现队列
 */
class MyQueue<T> 

    private Stack<T> stack1;
    private Stack<T> stack2;

    public MyQueue() 
        stack1 = new Stack<>();
        stack2 = new Stack<>();
    

    /**
     * 入队列
     */
    public void addLast(T val) 
        // 先将stack1的数据全部转移到stack2
        while (!stack1.isEmpty()) 
            stack2.push(stack1.pop());
        
        // 将val压入stack2栈顶
        stack2.push(val);
    

    /**
     * 出队列
     */
    public T pollFirst() 
        while (!stack2.isEmpty()) 
            // 将第二个栈里的数据从栈顶开始弹出,然后装入到stack1中
            stack1.push(stack2.pop());
        
        return stack1.pop();
    

    /**
     * 获取队列头元素
     */
    public T peekFirst() 
        while (!stack2.isEmpty()) 
            // 将第二个栈里的数据从栈顶开始弹出,然后装入到stack1中
            stack1.push(stack2.pop());
        
        return stack1.peek();
    

    public boolean isEmpty() 
        return stack1.isEmpty() && stack2.isEmpty();
    


测试代码

public static void main(String[] args) 
        MyQueue<String> myQueue = new MyQueue<>();
        myQueue.addLast("wenpan");
        myQueue.addLast("hello");
        myQueue.addLast("world");
        myQueue.addLast("1");
        myQueue.addLast("2");
        myQueue.addLast("3");
        for (int i = 0; i < 5; i++) 
            String str = myQueue.pollFirst();
            System.out.println(str);
        
        myQueue.addLast("4");
        myQueue.addLast("5");
        while (!myQueue.isEmpty()) 
            String str = myQueue.pollFirst();
            System.out.println(str);
        
    

输出

wenpan
hello
world
1
2
3
4
5

以上是关于java数据结构与算法之使用两个栈实现队列的主要内容,如果未能解决你的问题,请参考以下文章

PHP数据结构与算法之PHP实现队列栈

PHP数据结构与算法之PHP实现队列栈

数据结构之栈和队列及其Java实现

JAVA数据结构与算法之数组与队列

Leetcode题解——数据结构之栈和队列

黑马程序员 C语言数据结构与算法之线性表(链表/栈/队列/顺序表)