剑指offer E7 两个栈实现队列-CSDN博客

本文链接：https://blog.csdn.net/qq_32939135/article/details/122994443

该篇博客展示了如何利用两个栈实现一个队列以及如何用两个队列实现一个栈。在栈队互换的实现中，通过在适当的时候在两个栈之间转移元素来达到队列的appendTail和deleteHead操作，以及栈的push和pop操作。代码示例使用Java编写，详细解释了每个操作的过程。

摘要由CSDN通过智能技术生成

package com.algorithm.exersice;

import java.util.*;

/**
 * 用两个栈实现一个队列,实现appendTail和deleteHead
 * 用两个队列实现一个栈
 *
 */
public class E7 {
    public static void main(String[] args) {
        System.out.println("两个栈实现队列：");
        CQueue<Integer> queue = new CQueue<Integer>();
        queue.appendTail(1);
        queue.appendTail(2);
        queue.appendTail(3);
        System.out.println(queue.deleteHead());
        queue.appendTail(4);
        System.out.println(queue.deleteHead());
        System.out.println(queue.deleteHead());
        System.out.println(queue.deleteHead());
        System.out.println(queue.deleteHead());
        queue.appendTail(5);
        System.out.println(queue.deleteHead());

        System.out.println("两个队列实现栈：");
        CStack<Integer> stack = new CStack<>();
        stack.push(1);
        stack.push(2);
        System.out.println(stack.pop());
        stack.push(3);
        System.out.println(stack.pop());
        System.out.println(stack.pop());
        System.out.println(stack.pop());
    }

}

class CStack<T> {
    private Queue<T> queue1;
    private Queue<T> queue2;

    public CStack(){
        this.queue1 = new LinkedList<>();
        this.queue2 = new LinkedList<>();
    }

    public void push(T node) {
        queue1.add(node);
    }

    public T pop() {
        if (!queue1.isEmpty()) {
            //元素直接插入queue1，所以优先从queue1取出队尾元素
            T node = queue1.poll();
            while (!queue1.isEmpty()) {
                queue2.add(node);
                node = queue1.poll();
            }
            return node;
        } else if (!queue2.isEmpty()){
            //queue1为空时，取出queue2队尾元素
            T node = queue2.poll();
            while (!queue2.isEmpty()) {
                queue1.add(node);
                node = queue2.poll();
            }
            return node;
        } else {
            return null;
        }
    }
}

class CQueue<T> {
    private Stack<T> stack1;
    private Stack<T> stack2;

    public CQueue() {
        this.stack1 = new Stack<>();
        this.stack2 = new Stack<>();
    }

    public void appendTail(T node) {
        stack1.push(node);
    }

    public T deleteHead() {
        if (stack2.isEmpty()) {
            while (!stack1.isEmpty()) {
                stack2.push(stack1.pop());
            }
        }
        return stack2.isEmpty()? null :stack2.pop();
    }

}