【LeetCode】剑指09——用两个栈实现队列（难度：简单）

题目描述

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

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

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/yong-liang-ge-zhan-shi-xian-dui-lie-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

栈是先入后出的数据结构，为了实现队列的先入先出，需要用到另一个栈去改变栈中元素顺序，那么另一个栈的栈顶元素即为出队元素

代码详解

class CQueue {

    Stack<Integer> stack1; // 主要栈
    Stack<Integer> stack2; // 辅助栈
    boolean flag = true;   // true代表上次操作为入队，false代表上次操作为出队，初始值定为true

    public CQueue() { // 在构造函数中初始化两个栈
        stack1 = new Stack<>();
        stack2 = new Stack<>();
    }
    
    public void appendTail(int value) {
        if(!flag) { // 如果上次操作为出队，需要把stack2中元素移入stack1中
            while(!stack2.isEmpty()) {
                stack1.push(stack2.pop());
            }
        }
        // 上面的循环结束或者上次的操作就是入队，那么直接做在stack1做“入栈”，象征入队
        stack1.push(value);
        flag = true; // 本次操作为入队，flag状态改变为true
    }
    
    public int deleteHead() {
        if(flag) { // 如果上次操作为入队，需要把stack1中元素移入stack2中
            while(!stack1.isEmpty()) {
                stack2.push(stack1.pop());
            }
        }
        flag = false; // 本次操作为出队，flag状态改变为false
        if(stack2.isEmpty()) { // 如果stack2为空，象征队列空了，无法再出队任何元素
            return -1;
        }
        // 上面的循环结束或者上次的操作就是出队，那么直接做在stack2做“出栈”，象征出队
        return stack2.pop();
    }
}

/**
 * Your CQueue object will be instantiated and called as such:
 * CQueue obj = new CQueue();
 * obj.appendTail(value);
 * int param_2 = obj.deleteHead();
 */

注意点

• 为什么引入了一个flag标志位？

  如果上次和本次的操作均为入队/出队，那么无需在两个栈间执行无意义的元素移动，直接在对应的stack上执行入队/出队操作即可