剑指 Offer 09. 用两个栈实现队列

Posted 2022-01-06 Spring-_-Bear

用两个栈实现一个队列。队列的声明如下，请实现它的两个函数 appendTail 和 deleteHead ，分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素，deleteHead 操作返回 -1 )

示例 1：

输入：
[“CQueue”,“appendTail”,“deleteHead”,“deleteHead”]
[[],[3],[],[]]
输出：[null,null,3,-1]
示例 2：

输入：

[“CQueue”,“deleteHead”,“appendTail”,“appendTail”,“deleteHead”,“deleteHead”]
[[],[],[5],[2],[],[]]
输出：[null,-1,null,null,5,2]

提示：

1 <= values <= 10000
最多会对 appendTail、deleteHead 进行 10000 次调用

题解：

import java.util.Stack;

/**
 * 剑指 Offer 09. 用两个栈实现队列
 *
 * @author Spring-_-Bear
 * @date 2021-12-14 22:49
 */
public class CQueue 
    /*
     * 思路分析：题目要求使用两个栈实现队列，栈具有后进先出的特点，而队列具有先进先出的特点
     * 需要实现队列插入和删除元素操作即先插入的先删除，维护两个栈，第一个栈支持插入操作，
     * 第二栈支持删除操作，当需要删除元素的时候，如果 stack2 为空，则将 stack1 里的所有元素弹出插入到 stack2 里
     * 如果 stack2 仍为空，则返回 -1，否则从 stack2 弹出一个元素并返回
     */
    private Stack<Integer> stack1;
    private Stack<Integer> stack2;

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

    public void appendTail(int value) 
        stack1.push(value);
    

    public int deleteHead() 
        // 第二个栈为空，将第一个栈中的所有元素弹出压入到第二个栈中
        if (stack2.isEmpty()) 
            while (!stack1.isEmpty()) 
                stack2.push(stack2.pop());
            
        
        // 如果栈 2 不为空则返回栈顶元素；为空则返回 -1
        if (stack2.isEmpty()) 
            return -1;
         else 
            return stack2.pop();