JZ5 用两个栈实现队列

原题链接

描述

用两个栈来实现一个队列，分别完成在队列尾部插入整数(push)和在队列头部删除整数(pop)的功能。 队列中的元素为int类型。保证操作合法，即保证pop操作时队列内已有元素。

示例

输入：["PSH1","PSH2","POP","POP"]

返回值：1,2

思路

把stack1用于push，stack2用于pop达到队列FIFO的效果。没有限制两个栈的大小，所以第一次pop时直接把stack1的pop顺序push进stack2，然后stack2进行pop即可。

但是第二次pop时，就不能再像刚才那样了，因为stack2如果还有元素的话，说明这些元素都比stack1栈底的元素入站早。所以再有pop时要把stack2的元素顺序pop完，才能输出stack1的元素。

解答

import java.util.Stack;

public class Solution {
    Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 = new Stack<Integer>();
    
    public void push(int node) {
        stack1.push(node);
    }
    
    public int pop() {
        if(stack2.empty()){
            while(!stack1.empty()){
                stack2.push(stack1.pop());
            }
            return stack2.pop();
        }
        return stack2.pop();
    }
}

更

重做一遍力扣上的。用LinkedList 模拟 Stack。LinkedList实现了Deque接口，其 add() 方法是使用尾插法，removeLast() 方法也是删除尾部结点，刚好符合栈FILO的特性。

import java.util.LinkedList;

class CQueue {
    LinkedList<Integer> stackA,stackB;

    public CQueue() {
        stackA=new LinkedList<>();
        stackB=new LinkedList<>();
    }

    public void appendTail(int value) {
        stackA.add(value);
    }

    public int deleteHead() {
        if (stackB.isEmpty()){
            while (!stackA.isEmpty()){
                stackB.add(stackA.removeLast());
            }
        }
        return stackB.isEmpty()?-1:stackB.removeLast();
    }
}