Leetcode——用栈实现队列

Posted 2021-07-11 Yawn,

1. 题目

无论是用队列实现栈还是用栈实现队列，其实方法都一致：使得新插入的数处于队列头即可

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

2. 题解

解法二：双队列
用A队列保存栈内的元素，用B队列作为入栈的辅助队列

class MyStack {
    Deque<Integer> A,B; 

    public MyStack() {
        A = new LinkedList<>();
        B = new LinkedList<>();
    }
    
    public void push(int x) {
        B.offer(x); //先将要插入的元素加入到辅助队列中
        while (!A.isEmpty()) {  //然后再将A队列的所有元素加入到B队列，使得新插入的元素插入到了队列头
            B.offer(A.poll());
        }
        Deque<Integer> temp = A;    //此时交换A，B队列
        A = B;
        B = temp;
    }
    
    public int pop() {
        return A.poll();    //因为A队列队头保存着栈顶元素，所以直接出队就行
    }
    
    public int top() {
        return A.peek();    //因为A队列队头保存着栈顶元素,所以直接看队列头就可以
    }

    public boolean empty() {    //A保存着栈，所以队列A空，则栈肯定是空的
        return A.isEmpty();
    }
}

解法二：单队列
其实不难发现在入栈操作时，我们其实可以只用一个队列就能实现插入的数插入到队列头，并且保持相对顺序不变。

class MyStack {
    Deque<Integer> A; 

    public MyStack() {
        A = new LinkedList<>();
    }
    
    public void push(int x) {
        int n = A.size();   //先记录没插入之前队列的长度
        A.offer(x);         //插入到队列尾
        for(int i = 0; i < n; i++)  //循环移动使得队列尾移动到队列头
            A.offer(A.poll());
    }
    
    public int pop() {
        return A.poll();
    }
    
    public int top() {
        return A.peek();
    }
    
    public boolean empty() {
        return A.isEmpty();
    }
}