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

题目

用两个栈实现一个队列。队列的声明如下，请实现它的两个函数 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]

如果题目看不懂的话，可以看这道题的评论，有许多大佬讲的还是很清楚的，这里就不说了。

思路

题目说是用两个栈实现一个队列，首先得知道

栈：先进后出
队列：先进先出

所以，我理解的尾插操作为用两个栈实现队列的先进先出。

首先建立两个栈,其长度为0，表示其为空

Stack<Integer> stack1 = new Stack<Integer>();
Stack<Integer> stack2 =  new Stack<Integer>();
int size = 0;

由于栈是先进后出，所以每插入一个元素，需要将其他的元素都先移走，将尾插进去的元素放进去，再将移走的元素按之前的顺序放进去

  while(!stack1.isEmpty()){
        //将stack1中的元素出栈，再存入栈stack2中
            stack2.push(stack1.pop());
        }
        
        //此时栈stack1为空，然后插入要插入的元素，此时他就在栈底
        stack1.push(value);
        
          while(!stack2.isEmpty()){
        //将stack2中的元素出栈再存入栈stack1中，因为栈是先进后出，所以他的顺序还是保持原来的顺序
            stack1.push(stack2.pop());
        }
        //因为新加了一个元素，所以长度要增加
        size++;
    }

至于删除首部元素,由于他插入元素已经是按照她原来的顺序，所以栈顶元素就是首部元素，只需要将这个元素取出返回并使长度减一就行。

public int deleteHead() {
        if(size == 0){
            return -1;
        }
        int ret = stack1.pop();
        size--;
        return ret;
    }

代码

class CQueue {
//栈：先进后出；   队列：先进先出
    Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 =  new Stack<Integer>();
    int size = 0;

    public CQueue() {

    }
    
    public void appendTail(int value) {
        //尾插
        //stack1 底部保存新插入的元素，顶部保存他原有的元素
        while(!stack1.isEmpty()){
        //将stack1中的元素出栈，再存入栈stack2中
            stack2.push(stack1.pop());
        }
        //此时栈stack1为空，然后插入要插入的元素，此时他就在栈底
        stack1.push(value);
        

        while(!stack2.isEmpty()){
        //将stack2中的元素出栈再存入栈stack1中，因为栈是先进后出，所以他的顺序还是保持原来的顺序
            stack1.push(stack2.pop());
        }
        //因为新加了一个元素，所以长度要增加
        size++;
    }
    
    public int deleteHead() {
        if(size == 0){
            return -1;
        }
        int ret = stack1.pop();
        size--;
        return ret;
    }
}

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